We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 68431b8 commit def3ec7Copy full SHA for def3ec7
1 file changed
et_replay/tools/et_replay.py
@@ -43,7 +43,6 @@
43
from param_bench.train.compute.python.lib.init_helper import load_modules
44
from param_bench.train.compute.python.workloads import pytorch as workloads_pytorch
45
from torch._inductor.async_compile import AsyncCompile
46
-from torch._inductor.codecache import TritonFuture
47
48
# grid and split_scan_grid are dynamically loaded
49
from torch._inductor.runtime.triton_heuristics import grid, split_scan_grid # noqa
@@ -1038,8 +1037,6 @@ def _generate_run_ops_str(override):
1038
1037
func, output_count = self.funcs[node.id]
1039
if not func:
1040
continue
1041
- if isinstance(func, TritonFuture):
1042
- func = func.result()
1043
1044
func_str = f"funcs[{node.id}]"
1045
inputs_str = _generate_inputs_str(node)
0 commit comments