Make the tensors contiguous.

nod-ai · Jun 26, 2023 · 74a7202 · 74a7202
1 parent 27a0873
commit 74a7202
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/shark/dynamo_backend/utils.py b/shark/dynamo_backend/utils.py
@@ -136,7 +136,7 @@ def compile(self):
         self.shark_module = shark_module
 
     def __call__(self, *inputs):
-        np_inputs = [x.detach().cpu().numpy() for x in inputs]
+        np_inputs = [x.contiguous().detach().cpu().numpy() for x in inputs]
         np_outs = self.shark_module("forward", np_inputs)
         if self.was_unwrapped:
             np_outs = [