[PACKAGING] Added FileCheck

test/CMakeLists.txt

@@ -10,8 +10,8 @@ configure_lit_site_cfg(
 )
 
 set(TRITON_TEST_DEPENDS
-  # FileCheck count not
   triton-opt
+  FileCheck
 )
 
 add_lit_testsuite(check-triton "Running the triton regression tests"

test/Conversion/ops.mlir

@@ -1,4 +1,4 @@
-// RUN triton-opt %s -convert-triton-to-tritongpu
+// RUN: triton-opt %s -convert-triton-to-tritongpu
 
 func @ops() {
   %a = arith.constant dense<1.00e+00> : tensor<128x32xf16>

test/TritonGPU/ops.mlir

@@ -1,5 +0,0 @@
-// RUN: triton-opt %s -tritongpu-verifier
-func @test_dot(%a : tensor<128x32xf16>, %b : tensor<32x128xf16>, %c : tensor<128x128xf16>) {
-  %d = tt.dot %a, %b, %c {allowTF32 = true} : tensor<128x32xf16> * tensor<32x128xf16> -> tensor<128x128xf16>
-  return
-}

test/lit.cfg.py

@@ -48,11 +48,12 @@ config.test_source_root = os.path.dirname(__file__)
 # test_exec_root: The root path where tests should be run.
 config.test_exec_root = os.path.join(config.triton_obj_root, 'test')
 config.triton_tools_dir = os.path.join(config.triton_obj_root, 'bin')
+config.filecheck_dir = os.path.join(config.triton_obj_root, 'bin', 'FileCheck')
+tool_dirs = [config.triton_tools_dir, config.llvm_tools_dir, config.filecheck_dir]
 
 # Tweak the PATH to include the tools dir.
-llvm_config.with_environment('PATH', config.llvm_tools_dir, append_path=True)
-
-tool_dirs = [config.triton_tools_dir, config.llvm_tools_dir]
+for d in tool_dirs:
+    llvm_config.with_environment('PATH', d, append_path=True)
 tools = [
     'triton-opt',
     ToolSubst('%PYTHON', config.python_executable, unresolved='ignore'),