Raise error for dot int64 input (#130)

abhudev · kulinseth · commit 1b32ddcb536f · 2022-12-09T00:13:58.000-08:00
diff --git a/aten/src/ATen/native/mps/operations/Blas.mm b/aten/src/ATen/native/mps/operations/Blas.mm
@@ -21,6 +21,9 @@ Tensor dot_mps(
   const Tensor &self,
   const Tensor &other)
 {
+
+  TORCH_CHECK(self.scalar_type() != ScalarType::Long, "MPS: dot op doesn't support int64 input")
+
   using namespace mps;
   auto output = at::native::empty_mps({}, self.scalar_type(), c10::nullopt, kMPS, c10::nullopt, c10::nullopt);
 

Original file line number	Diff line number	Diff line change
`@@ -21,6 +21,9 @@ Tensor dot_mps(`
`21`	`21`	`const Tensor &self,`
`22`	`22`	`const Tensor &other)`
`23`	`23`	`{`
	`24`	`+`
	`25`	`+ TORCH_CHECK(self.scalar_type() != ScalarType::Long, "MPS: dot op doesn't support int64 input")`
	`26`	`+`
`24`	`27`	`using namespace mps;`
`25`	`28`	`auto output = at::native::empty_mps({}, self.scalar_type(), c10::nullopt, kMPS, c10::nullopt, c10::nullopt);`
`26`	`29`