diff options
author | Fabian Mora <fmora.dev@gmail.com> | 2023-09-14 18:00:27 -0400 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-09-14 18:00:27 -0400 |
commit | 5093413a5007b017a530edbeed42d32bfd18b126 (patch) | |
tree | f5e8406133c831911222cd1759a1ae2b46705235 /mlir/test/lit.site.cfg.py.in | |
parent | 6f4a5286981dd57826751834f7a434be1a463339 (diff) | |
download | llvm-5093413a5007b017a530edbeed42d32bfd18b126.zip llvm-5093413a5007b017a530edbeed42d32bfd18b126.tar.gz llvm-5093413a5007b017a530edbeed42d32bfd18b126.tar.bz2 |
[mlir][gpu][NVPTX] Enable NVIDIA GPU JIT compilation path (#66220)
This patch adds an NVPTX compilation path that enables JIT compilation
on NVIDIA targets. The following modifications were performed:
1. Adding a format field to the GPU object attribute, allowing the
translation attribute to use the correct runtime function to load the
module. Likewise, a dictionary attribute was added to add any possible
extra options.
2. Adding the `createObject` method to `GPUTargetAttrInterface`; this
method returns a GPU object from a binary string.
3. Adding the function `mgpuModuleLoadJIT`, which is only available for
NVIDIA GPUs, as there is no equivalent for AMD.
4. Adding the CMake flag `MLIR_GPU_COMPILATION_TEST_FORMAT` to specify
the format to use during testing.
Diffstat (limited to 'mlir/test/lit.site.cfg.py.in')
-rw-r--r-- | mlir/test/lit.site.cfg.py.in | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/mlir/test/lit.site.cfg.py.in b/mlir/test/lit.site.cfg.py.in index ef1fdbc..2de40ba 100644 --- a/mlir/test/lit.site.cfg.py.in +++ b/mlir/test/lit.site.cfg.py.in @@ -29,6 +29,7 @@ config.run_cuda_tests = @MLIR_ENABLE_CUDA_CONVERSIONS@ config.enable_cuda_runner = @MLIR_ENABLE_CUDA_RUNNER@ config.run_rocm_tests = @MLIR_ENABLE_ROCM_CONVERSIONS@ config.enable_rocm_runner = @MLIR_ENABLE_ROCM_RUNNER@ +config.gpu_compilation_format = "@MLIR_GPU_COMPILATION_TEST_FORMAT@" config.rocm_test_chipset = "@ROCM_TEST_CHIPSET@" config.enable_spirv_cpu_runner = @MLIR_ENABLE_SPIRV_CPU_RUNNER@ config.enable_vulkan_runner = @MLIR_ENABLE_VULKAN_RUNNER@ |