pytorch
diff --git a/‎aten/src/ATen/native/mps/OperationUtils.mm‎
Lines changed: 1 addition & 1 deletion b/‎aten/src/ATen/native/mps/OperationUtils.mm‎
Lines changed: 1 addition & 1 deletion
@@ -404,7 +404,7 @@ void resize_tensor(Tensor* output) {
 // this is meant to suppress the availability warning on castTensor
 // we pass ScalarType instead of MPSDataType to handle MPSDataTypeBoolean's availability too
 MPSGraphTensor* castMPSTensor(MPSGraph *mpsGraph, MPSGraphTensor* tensor, MPSDataType toType) {
-  if ([tensor dataType] == toType) {
+  if (!tensor || [tensor dataType] == toType) {
     return tensor;
   }
   return [mpsGraph castTensor:tensor toType:toType name:@"castTensor"];
Original file line number	Diff line number	Diff line change
`@@ -404,7 +404,7 @@ void resize_tensor(Tensor* output) {`
`404`	`404`	`// this is meant to suppress the availability warning on castTensor`
`405`	`405`	`// we pass ScalarType instead of MPSDataType to handle MPSDataTypeBoolean's availability too`
`406`	`406`	`MPSGraphTensor* castMPSTensor(MPSGraph mpsGraph, MPSGraphTensor tensor, MPSDataType toType) {`
`407`		`- if ([tensor dataType] == toType) {`
	`407`	`+ if (!tensor \|\| [tensor dataType] == toType) {`
`408`	`408`	`return tensor;`
`409`	`409`	`}`
`410`	`410`	`return [mpsGraph castTensor:tensor toType:toType name:@"castTensor"];`