Skip to content

Commit 14e589e

Browse files
Remove env var TRITON_INTEL_ENABLE_BLOCK_IO_STORE_ON_REGULAR_PTR (#4851)
There was no regression discovered after enabling `TRITON_INTEL_ENABLE_BLOCK_IO_STORE_ON_REGULAR_PTR` by default. Since we always want to enable this behavior, the env var is not needed. Signed-off-by: Whitney Tsang <[email protected]>
1 parent 0ca9cda commit 14e589e

File tree

4 files changed

+2
-9
lines changed

4 files changed

+2
-9
lines changed

include/triton/Tools/Sys/GetEnv.hpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,7 +48,6 @@ inline const std::set<std::string> CACHE_INVALIDATING_ENV_VARS = {
4848
"TRITON_INTEL_AGGRESSIVE_DPAS_REUSE",
4949
"TRITON_INTEL_DO_NOT_SINK_INSTR_ACROSS_RGN",
5050
"TRITON_INTEL_ENABLE_BLOCK_IO_ALL_LAYOUTS",
51-
"TRITON_INTEL_ENABLE_BLOCK_IO_STORE_ON_REGULAR_PTR",
5251
"TRITON_INTEL_ENABLE_FIRST_LOAD_TO_SLM",
5352
"TRITON_INTEL_ENABLE_INSTR_SCHED",
5453
"TRITON_INTEL_FAST_MATH",

python/test/unit/intel/test_block_store.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
import triton
1010
from triton._internal_testing import is_xpu
1111

12-
os.environ["TRITON_INTEL_ENABLE_BLOCK_IO_STORE_ON_REGULAR_PTR"] = "1"
1312
os.environ["TRITON_INTEL_ENABLE_BLOCK_IO_ALL_LAYOUTS"] = "1"
1413

1514

test/TritonIntelGPU/tensor-pointer-store-block-2d.mlir

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
1-
// RUN: env TRITON_INTEL_ENABLE_BLOCK_IO_STORE_ON_REGULAR_PTR=1 triton-opt %s -split-input-file --intel-allocate-shared-memory --convert-triton-intel-gpu-to-llvm | FileCheck %s --implicit-check-not=llvm.inline_asm
2-
// RUN: env TRITON_INTEL_ENABLE_BLOCK_IO_STORE_ON_REGULAR_PTR=1 TRITON_INTEL_ENABLE_BLOCK_IO_ALL_LAYOUTS=1 triton-opt %s -split-input-file --intel-allocate-shared-memory --convert-triton-intel-gpu-to-llvm | FileCheck %s --implicit-check-not=llvm.inline_asm --check-prefixes=ALL-LAYOUT
1+
// RUN: triton-opt %s -split-input-file --intel-allocate-shared-memory --convert-triton-intel-gpu-to-llvm | FileCheck %s --implicit-check-not=llvm.inline_asm
2+
// RUN: env TRITON_INTEL_ENABLE_BLOCK_IO_ALL_LAYOUTS=1 triton-opt %s -split-input-file --intel-allocate-shared-memory --convert-triton-intel-gpu-to-llvm | FileCheck %s --implicit-check-not=llvm.inline_asm --check-prefixes=ALL-LAYOUT
33

44
#blocked = #ttg.blocked<{sizePerThread = [1, 4, 2], threadsPerWarp = [1, 1, 32], warpsPerCTA = [1, 8, 2], order = [2, 1, 0]}>
55
#slice = #ttg.slice<{dim = 1, parent = #blocked}>

third_party/intel/lib/TritonIntelGPUToLLVM/LoadStoreOpToLLVM.cpp

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2772,11 +2772,6 @@ struct StoreOpToBlockIOConversion
27722772
offsetBaseX = offsetX;
27732773
offsetBaseY = offsetY;
27742774
} else {
2775-
std::optional<bool> enableBlockStore =
2776-
mlir::triton::tools::isEnvValueBool(mlir::triton::tools::getStrEnv(
2777-
"TRITON_INTEL_ENABLE_BLOCK_IO_STORE_ON_REGULAR_PTR"));
2778-
if (enableBlockStore.has_value() && !enableBlockStore.value())
2779-
return failure();
27802775
// Get the LLVM values for pointers
27812776
ptrElems = unpackLLElements(loc, llPtr, rewriter);
27822777
assert(ptrElems.size() == numElems &&

0 commit comments

Comments
 (0)