Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 0 additions & 3 deletions et_replay/tools/et_replay.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,6 @@
from param_bench.train.compute.python.lib.init_helper import load_modules
from param_bench.train.compute.python.workloads import pytorch as workloads_pytorch
from torch._inductor.async_compile import AsyncCompile
from torch._inductor.codecache import TritonFuture

# grid and split_scan_grid are dynamically loaded
from torch._inductor.runtime.triton_heuristics import grid, split_scan_grid # noqa
Expand Down Expand Up @@ -1038,8 +1037,6 @@ def _generate_run_ops_str(override):
func, output_count = self.funcs[node.id]
if not func:
continue
if isinstance(func, TritonFuture):
func = func.result()

func_str = f"funcs[{node.id}]"
inputs_str = _generate_inputs_str(node)
Expand Down