Skip to content

Commit 48949cb

Browse files
razarmehrkulinseth
authored andcommitted
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

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

aten/src/ATen/mps/MPSStream.h

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -34,8 +34,6 @@ typedef void* MTLDevice_t;
3434
namespace at {
3535
namespace mps {
3636

37-
#define USE_MPSCOMMANDBUFFER 1
38-
3937
//-----------------------------------------------------------------
4038
// MPSStream
4139
//-----------------------------------------------------------------

0 commit comments

Comments
 (0)