Fix empty dispatch for bitwise_not (#132)

kulinseth · Oct 31, 2022 · 77bdce7 · 77bdce7
1 parent 897e314
commit 77bdce7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/aten/src/ATen/native/mps/operations/BitwiseOps.mm b/aten/src/ATen/native/mps/operations/BitwiseOps.mm
@@ -313,6 +313,11 @@ void handle_tensor_scalar_binary_op(const at::Tensor& self, const at::Scalar& ot
  getMetalType(self),
  getMetalType(self),
  "bitwise_not");
+ uint32_t length = output.numel();
+ if (length == 0) {
+ return output_;
+ }
+
  dispatch_sync(stream->queue(), ^(){
  id<MTLCommandBuffer> buffer = stream->commandBuffer();
  id<MTLComputeCommandEncoder> commandEncoder = [buffer computeCommandEncoder];