Compute node analysis (facebookresearch#127)

Summary: Pull Request resolved: facebookresearch#127 Compute node analysis Differential Revision: D58610032
shengfukevin · Jun 25, 2024 · dcd020a · dcd020a
1 parent b41079e
commit dcd020a
Show file tree

Hide file tree

Showing 2 changed files with 138 additions and 136 deletions.
diff --git a/et_replay/lib/et_replay_utils.py b/et_replay/lib/et_replay_utils.py
@@ -464,8 +464,7 @@ def build_triton_func(n, resources_dir, async_compile, device):
         func = async_compile.triton(n.name, code, device_str=device)
     except Exception:
         func = async_compile.triton("triton_", code, device_str=device)
-    finally:
-        func = None
+
     return func, 0