Skip to content

Commit 08f881e

Browse files
authored
Fix the uint8 type issue with View ops kernels (#348)
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 b520970 commit 08f881e

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -696,7 +696,7 @@ static IntArrayRef updateTensorBaseShape(const Tensor& self)
696696
{c10::ScalarType::Int, "int"},
697697
{c10::ScalarType::Short, "short"},
698698
{c10::ScalarType::Char, "char"},
699-
{c10::ScalarType::Byte, "char"},
699+
{c10::ScalarType::Byte, "uchar"},
700700
{c10::ScalarType::Bool, "bool"},
701701
};
702702

0 commit comments

Comments
 (0)