add experimental env for disabling llvm slpvectorization

Summary: Add an experimental flag to disable slpvectorization for helion testing

Reviewed By: agron911

Differential Revision: D89382448

fbshipit-source-id: bf5f5f09e13bcbd6cf757ad4c000d1f1026c9d93

Author: CRobeck

include/triton/Tools/Sys/GetEnv.hpp

@@ -30,6 +30,7 @@ inline const std::set<std::string> CACHE_INVALIDATING_ENV_VARS = {
     "MLIR_DISABLE_MULTITHREADING",
     "TRITON_DEFAULT_FP_FUSION",
     "TRITON_DISABLE_LINE_INFO",
+    "TRITON_DISABLE_SLPVECTORIZATION",
     "TRITON_ENABLE_LLVM_DEBUG",
     "TRITON_HIP_GLOBAL_PREFETCH",
     "TRITON_HIP_LOCAL_PREFETCH",

python/src/llvm.cc

@@ -354,6 +354,13 @@ void init_triton_llvm(py::module &&m) {
         // regressions with some scheduling solution.
         tuningOptions.SLPVectorization = true;
 
+        bool disableSLPVectorization =
+            mlir::triton::tools::getBoolEnv("TRITON_DISABLE_SLPVECTORIZATION");
+
+        if (disableSLPVectorization) {
+          tuningOptions.SLPVectorization = false;
+        }
+
         std::string pluginFile =
             mlir::triton::tools::getStrEnv("LLVM_PASS_PLUGIN_PATH");