Change commitAndWait to Commit in View Ops (#73)

kulinseth · Jul 21, 2022 · 9319b02 · 9319b02
1 parent e68dcc5
commit 9319b02
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/aten/src/ATen/mps/MPSStream.mm b/aten/src/ATen/mps/MPSStream.mm
@@ -45,7 +45,7 @@
  // typically in GPU to GPU copies we won't commit explicitly
  break;
  case SyncType::COMMIT:
- commit(true);
+ flush();
  break;
  case SyncType::COMMIT_AND_WAIT:
  commitAndWait();

diff --git a/aten/src/ATen/native/mps/operations/View.mm b/aten/src/ATen/native/mps/operations/View.mm
@@ -73,7 +73,7 @@
  cachedGraph->outputTensor : outputTensorData
  };
  stream->executeMPSGraph(cachedGraph->graph(), feeds, results,
- requires_sync ? SyncType::COMMIT_AND_WAIT : SyncType::NONE);
+ requires_sync ? SyncType::COMMIT : SyncType::NONE);
  }
  return output;
 }
@@ -240,7 +240,7 @@ Tensor gatherViewTensor(const at::Tensor& src, at::Tensor& dst)
 {
  ViewCachedGraph* cachedGraph = createViewGraph(output, output.sizes(), output.strides(),
  output.storage_offset(), /*needsScatter*/ true);
- return runViewGraph(cachedGraph, src, output, /*needsScatter*/ true);
+ return runViewGraph(cachedGraph, src, output, /*needsScatter*/ true, /*requires_sync*/ true);
 }
 
 } // namespace mps