diff --git a/python/test/unit/tools/test_disasm.py b/python/test/unit/tools/test_disasm.py index 7001d314f9..8177482813 100644 --- a/python/test/unit/tools/test_disasm.py +++ b/python/test/unit/tools/test_disasm.py @@ -34,4 +34,4 @@ def kernel(X, i: tl.constexpr): assert x[0] == 12 dis = h.asm["spvdis"] # check that the spvdis has a store instruction. - assert "PredicatedStore" in dis + assert "OpStore" in dis diff --git a/third_party/intel/lib/TritonIntelGPUToLLVM/LoadStoreOpToLLVM.cpp b/third_party/intel/lib/TritonIntelGPUToLLVM/LoadStoreOpToLLVM.cpp index 6c1dcd9d22..4fc0178dae 100644 --- a/third_party/intel/lib/TritonIntelGPUToLLVM/LoadStoreOpToLLVM.cpp +++ b/third_party/intel/lib/TritonIntelGPUToLLVM/LoadStoreOpToLLVM.cpp @@ -3401,10 +3401,7 @@ struct StoreOpConversion if (maskVal) { // Create a predicated store operation. - std::optional enablePredicated = - mlir::triton::tools::isEnvValueBool( - mlir::triton::tools::getStrEnv("TRITON_INTEL_PREDICATED")); - if (!enablePredicated.has_value() || enablePredicated.value()) + if (triton::tools::getBoolEnv("TRITON_INTEL_PREDICATED")) rewriter.create( loc, addrElem, vecWord, b.i64_val(alignment), maskVal); else