File tree Expand file tree Collapse file tree 1 file changed +0
-3
lines changed Expand file tree Collapse file tree 1 file changed +0
-3
lines changed Original file line number Diff line number Diff line change 4343from param_bench .train .compute .python .lib .init_helper import load_modules
4444from param_bench .train .compute .python .workloads import pytorch as workloads_pytorch
4545from torch ._inductor .async_compile import AsyncCompile
46- from torch ._inductor .codecache import TritonFuture
4746
4847# grid and split_scan_grid are dynamically loaded
4948from torch ._inductor .runtime .triton_heuristics import grid , split_scan_grid # noqa
@@ -1038,8 +1037,6 @@ def _generate_run_ops_str(override):
10381037 func , output_count = self .funcs [node .id ]
10391038 if not func :
10401039 continue
1041- if isinstance (func , TritonFuture ):
1042- func = func .result ()
10431040
10441041 func_str = f"funcs[{ node .id } ]"
10451042 inputs_str = _generate_inputs_str (node )
You can’t perform that action at this time.
0 commit comments