Skip to content

Commit 2dfb6d3

Browse files
committed
[intel] enable 'test_line_info_env'
Signed-off-by: Anatoly Myachev <[email protected]>
1 parent fd28ab7 commit 2dfb6d3

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

python/test/unit/language/test_line_info.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -266,5 +266,8 @@ def test_line_info_env(monkeypatch, status: str):
266266
monkeypatch.setenv("TRITON_DISABLE_LINE_INFO", status)
267267
kernel_single.device_caches.clear()
268268
kernel_info = kernel_single.warmup(torch.float32, torch.float32, BLOCK=shape[0], grid=(1, ))
269-
file_lines = extract_file_lines(command, anchor, separator, kernel_info.asm[obj_kind])
269+
if obj_kind == "spvbin":
270+
file_lines = spv_extract_file_lines(kernel_info.asm["spv"], command)
271+
else:
272+
file_lines = extract_file_lines(command, anchor, separator, kernel_info.asm[obj_kind])
270273
assert len(file_lines) == 0 if status == "1" else len(file_lines) > 0

0 commit comments

Comments
 (0)