make --dump-ir also dump sass (#2460)

Summary:
Triton has a `get_sass` utility that disassembles the cubin and gets the sass - in this PR, we try to use `get_sass` when running with `--dump-ir`, if we can find a cubin file.

Pull Request resolved: #2460

Reviewed By: xuzhao9

Differential Revision: D62925358

Pulled By: davidberard98

fbshipit-source-id: 7aa1e66c43222b776c949ced3fdcabcee30fbb21

torchbenchmark/util/triton_op.py

@@ -1155,3 +1155,11 @@ def run_and_capture(self, *args, **kwargs):
                         ir_dir / f"{fn._name}_{kernel.name}_{input_id}.{ir}", "w"
                     ) as f:
                         f.write(kernel.asm[ir])
+            if "cubin" in kernel.asm:
+                from triton.tools.disasm import get_sass
+
+                sass = get_sass(kernel.asm["cubin"])
+                with open(
+                    ir_dir / f"{fn._name}_{kernel.name}_{input_id}.sass", "w"
+                ) as f:
+                    f.write(sass)