Skip to content

Commit 913ea6b

Browse files
author
Ian Schweer
committed
Add super call, disable compiler in tests
1 parent 8fd37d0 commit 913ea6b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/link/pytorch/test_blockwise.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@ class TestOp(Op):
1616
gufunc_signature = "(m,n),(n,p)->(m,p)"
1717

1818
def __init__(self, final_shape):
19+
super().__init__()
1920
self.final_shape = final_shape
2021
self.call_shapes = []
2122

@@ -28,6 +29,7 @@ def perform(self, *_):
2829

2930
@pytorch_funcify.register(TestOp)
3031
def evaluate_test_op(op, **_):
32+
@torch.compiler.disable(recursive=False)
3133
def func(a, b):
3234
op.call_shapes.extend(map(torch.Tensor.size, [a, b]))
3335
return a @ b

0 commit comments

Comments
 (0)