Skip to content

Commit 464ff18

Browse files
committed
Fix the uint8 type issue with View ops kernels
This should fix the problem in Resnet model with image artifacts due to saturation on int8 type and also the incorrect class recognition reported in pytorch#86954
1 parent bedeb1f commit 464ff18

File tree

1 file changed

+1
-1
lines changed
  • aten/src/ATen/native/mps/operations

1 file changed

+1
-1
lines changed

aten/src/ATen/native/mps/operations/View.mm

+1-1
Original file line numberDiff line numberDiff line change
@@ -751,7 +751,7 @@ static IntArrayRef updateTensorBaseShape(const Tensor& self)
751751
{c10::ScalarType::Int, "int"},
752752
{c10::ScalarType::Short, "short"},
753753
{c10::ScalarType::Char, "char"},
754-
{c10::ScalarType::Byte, "char"},
754+
{c10::ScalarType::Byte, "uchar"},
755755
{c10::ScalarType::Bool, "bool"},
756756
};
757757

0 commit comments

Comments
 (0)