Skip to content

Commit 863e42f

Browse files
authored
[DEBUG] dump sass text when TRITON_KERNEL_DUMP is set (#7290)
1 parent 54606e8 commit 863e42f

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

python/triton/compiler/compiler.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -371,6 +371,9 @@ def compile(src, target=None, options=None):
371371
metadata_group[ir_filename] = fn_cache_manager.put(next_module, ir_filename)
372372
if fn_dump_manager is not None:
373373
fn_dump_manager.put(next_module, ir_filename)
374+
if ext == "cubin":
375+
sass = get_sass(next_module)
376+
fn_dump_manager.put(sass, file_name + ".sass")
374377
# use an env variable to parse ir from file
375378
if use_ir_loc == ext:
376379
ir_full_name = fn_cache_manager.get_file(ir_filename)

0 commit comments

Comments
 (0)