Merge pull request #253 from ndif-team/torch-patching

feat (api): add patching for torch.cat()
ndif-team · Oct 4, 2024 · b21ce8a · b21ce8a
2 parents cb991ff + 75afb82
commit b21ce8a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/nnsight/contexts/GraphBasedContext.py b/src/nnsight/contexts/GraphBasedContext.py
@@ -307,6 +307,7 @@ class GlobalTracingContext(GraphBasedContext):
             global_patch(torch, "randn"),
             global_patch(torch, "randperm"),
             global_patch(torch, "zeros"),
+            global_patch(torch, "cat")
         ]
         + [
             global_patch_class(value)