Supported programming languages for solution implementations. Enumeration of programming languages that can be used to implement solutions for computational workloads.
| 26 | |
| 27 | |
| 28 | class SupportedLanguages(str, Enum): |
| 29 | """Supported programming languages for solution implementations. |
| 30 | |
| 31 | Enumeration of programming languages that can be used to implement |
| 32 | solutions for computational workloads. |
| 33 | """ |
| 34 | |
| 35 | # Python languages |
| 36 | PYTORCH = "pytorch" |
| 37 | """PyTorch programming language. Specified if the solution uses PyTorch operators.""" |
| 38 | TRITON = "triton" |
| 39 | """Triton GPU programming language.""" |
| 40 | CUTE_DSL = "cute_dsl" |
| 41 | """NVIDIA CuTe DSL programming language.""" |
| 42 | CUTILE = "cutile" |
| 43 | """NVIDIA cuTile programming language.""" |
| 44 | CUDNN_FRONTEND = "cudnn_frontend" |
| 45 | """NVIDIA cuDNN frontend programming language.""" |
| 46 | # C++ languages |
| 47 | CUTLASS = "cutlass" |
| 48 | """Pure C++ source code.""" |
| 49 | CUDNN = "cudnn" |
| 50 | """NVIDIA cuDNN programming language.""" |
| 51 | CUBLAS = "cublas" |
| 52 | """NVIDIA cuBLAS programming language.""" |
| 53 | CUDA_CPP = "cuda_cpp" |
| 54 | """CUDA C++ programming language with inline PTX support.""" |
| 55 | |
| 56 | |
| 57 | class SupportedHardware(str, Enum): |
no outgoing calls