Commit 48949cb
Refactor blit copies and reuse scalar tensor's MPS storage in BinaryOps (#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>1 parent d03b2d7 commit 48949cb
1 file changed
+0
-2
lines changed| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
34 | 34 | | |
35 | 35 | | |
36 | 36 | | |
37 | | - | |
38 | | - | |
39 | 37 | | |
40 | 38 | | |
41 | 39 | | |
| |||
0 commit comments