Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Refactor blit copies and reuse scalar tensor's MPS storage in BinaryO…
…ps (#52) * Refactor blit copies and use commitAndContinue in commitAndWait() * Remove commitAndContinue from commitAndWait() * [MPS] Add commitAndContinue to non blocking blits. (#51) * Merge commitAndContinue changes from mps_master * Don't use getMPSGraphTensorFromScalar() if tensor is already on MPS device in Binary Ops This improves performance by preventing copy from GPU to CPU and back to GPU again Co-authored-by: Kulin Seth <kulin_seth@apple.com>
- Loading branch information