[BACKEND] Compile AMD backend (#722)

Co-authored-by: Whitney Tsang <whitney.tsang@intel.com>
intel · Mar 20, 2024 · b60101a · b60101a
1 parent 26a161d
commit b60101a
Show file tree

Hide file tree

Showing 4 changed files with 1 addition and 7 deletions.
diff --git a/python/setup.py b/python/setup.py
@@ -396,7 +396,7 @@ def build_extension(self, ext):
     f"https://anaconda.org/nvidia/cuda-nvdisasm/{version}/download/linux-{arch}/cuda-nvdisasm-{version}-0.tar.bz2",
 )
 
-backends = [*BackendInstaller.copy(["intel"]), *BackendInstaller.copy_externals()]
+backends = [*BackendInstaller.copy(["intel", "amd"]), *BackendInstaller.copy_externals()]
 
 
 def add_link_to_backends():

diff --git a/test/Conversion/amd/fp_to_fp.mlir b/test/Conversion/amd/fp_to_fp.mlir
@@ -1,6 +1,4 @@
 // RUN: triton-opt %s --split-input-file --convert-triton-amdgpu-to-llvm | FileCheck %s
-// FIXME: Currently unable to invoke amd passes.
-// XFAIL: *
 
 //  CHECK-LABEL: f16_to_f32
 #blocked = #triton_gpu.blocked<{sizePerThread = [1, 8], threadsPerWarp = [4, 8], warpsPerCTA = [4, 1], order = [1, 0]}>

diff --git a/test/Conversion/amd/tritongpu_wmma_dot_to_llvm.mlir b/test/Conversion/amd/tritongpu_wmma_dot_to_llvm.mlir
@@ -1,6 +1,4 @@
 // RUN: triton-opt %s --split-input-file --convert-triton-amdgpu-to-llvm | FileCheck %s
-// FIXME: Currently unable to invoke amd passes.
-// XFAIL: *
 
 //  CHECK-LABEL: wmma_dot
 #blocked = #triton_gpu.blocked<{sizePerThread = [1, 8], threadsPerWarp = [4, 8], warpsPerCTA = [4, 1], order = [1, 0]}>

diff --git a/test/TritonGPU/optimize_epilogue.mlir b/test/TritonGPU/optimize_epilogue.mlir
@@ -1,6 +1,4 @@
 // RUN: triton-opt %s -split-input-file --tritonamdgpu-optimize-epilogue | FileCheck --check-prefixes=GCN %s
-// FIXME: Unknown command line argument '--tritonamdgpu-optimize-epilogue'
-// XFAIL: *
 
 #mfma = #triton_gpu.amd_mfma<{warpsPerCTA=[1,1], instrShape=[32,32], isTranspose=false}>
 #blocked = #triton_gpu.blocked<{sizePerThread = [4, 4], threadsPerWarp = [4, 16], warpsPerCTA = [1, 1], order = [1, 0]}>