Skip to content

Commit aa01f5c

Browse files
authored
Temporarily disable predicated store (#5412)
This PR disables predicated store by default until #5411 is fixed. Signed-off-by: Anatoly Myachev <[email protected]>
1 parent b5d3396 commit aa01f5c

File tree

2 files changed

+2
-5
lines changed

2 files changed

+2
-5
lines changed

python/test/unit/tools/test_disasm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,4 +34,4 @@ def kernel(X, i: tl.constexpr):
3434
assert x[0] == 12
3535
dis = h.asm["spvdis"]
3636
# check that the spvdis has a store instruction.
37-
assert "PredicatedStore" in dis
37+
assert "OpStore" in dis

third_party/intel/lib/TritonIntelGPUToLLVM/LoadStoreOpToLLVM.cpp

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3401,10 +3401,7 @@ struct StoreOpConversion
34013401

34023402
if (maskVal) {
34033403
// Create a predicated store operation.
3404-
std::optional<bool> enablePredicated =
3405-
mlir::triton::tools::isEnvValueBool(
3406-
mlir::triton::tools::getStrEnv("TRITON_INTEL_PREDICATED"));
3407-
if (!enablePredicated.has_value() || enablePredicated.value())
3404+
if (triton::tools::getBoolEnv("TRITON_INTEL_PREDICATED"))
34083405
rewriter.create<TritonGEN::PredicatedStoreOp>(
34093406
loc, addrElem, vecWord, b.i64_val(alignment), maskVal);
34103407
else

0 commit comments

Comments
 (0)