search

uber / neuropod

A uniform interface to run deep learning models from multiple frameworks
https://neuropod.ai
Apache License 2.0
936 stars 77 forks source link

Ensure TorchScript output tensors are contiguous #554

Closed VivekPanyam closed 2 years ago

VivekPanyam commented 2 years ago

Summary:

Ensures that all tensors returned from a TorchScript model are contiguous.

Fixes #553

Test Plan:

Added a test. The test did not pass before the fix and it does pass after the fix.

codecov[bot] commented 2 years ago

Codecov Report

Merging #554 (0a02f55) into master (d62a357) will increase coverage by 0.00%. The diff coverage is 100.00%.

@@           Coverage Diff           @@
##           master     #554   +/-   ##
=======================================
  Coverage   86.56%   86.56%           
=======================================
  Files         105      105           
  Lines        7069     7070    +1     
=======================================
+ Hits         6119     6120    +1     
  Misses        950      950
Impacted Files Coverage Δ
...rce/neuropod/backends/torchscript/torch_backend.cc 81.31% <100.00%> (+0.04%) :arrow_up:

Continue to review full report at Codecov.

Legend - Click here to learn more Δ = absolute <relative> (impact), ø = not affected, ? = missing data Powered by Codecov. Last update d62a357...0a02f55. Read the comment docs.