github/SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/SHARK-Studio.git synced 2026-01-09 13:57:54 -05:00
Code Issues Packages Projects Releases Wiki Activity

Make the tensors contiguous.

Browse Source
This commit is contained in:
Prashant Kumar
2023-06-26 11:39:40 +00:00
parent 27a08735db
commit 74a7202173
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
shark/dynamo_backend/utils.py
View File

@@ -136,7 +136,7 @@ class SharkBackend:
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 = [