Skip to content

Commit af3b01b

Browse files
authored
Disable cache in test_line_info.py::test_line_info_ir_source (#5446)
Fixes #5444 by forcing TRITON_ALWAYS_COMPILE flag for the duration of the test. Signed-off-by: Sławomir Błauciak <slawomir.blauciak@intel.com>
1 parent 4c52491 commit af3b01b

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

python/test/unit/language/test_line_info.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -289,6 +289,7 @@ def test_line_info_ir_source(monkeypatch, status, tmp_path):
289289
#loc3 = loc("/path/test.py":9:4)
290290
"""
291291
monkeypatch.setenv("USE_IR_LOC", status)
292+
monkeypatch.setenv("TRITON_ALWAYS_COMPILE", "1")
292293
temp_file = tmp_path / "test.ttir"
293294
temp_file.write_text(src)
294295
kernel_info = triton.compile(str(temp_file))
@@ -314,7 +315,7 @@ def test_use_name_loc_as_prefix(fresh_triton_cache):
314315

315316
@triton.jit
316317
def kernel_basic(src, N, BLOCK_SIZE: tl.constexpr):
317-
# CHECK: #loc = loc("{{.*}}":316:0)
318+
# CHECK: #loc = loc("{{.*}}":317:0)
318319
# CHECK-LABEL: tt.func public @kernel_basic(
319320
# CHECK-SAME: %src: !tt.ptr<f32> loc("src"(#loc)), %N: i32 loc("N"(#loc)))
320321
# CHECK: %x_plus_1 = arith.constant dense<1.000000e+00> : tensor<16xf32> loc(#loc14)

0 commit comments

Comments
 (0)