Skip to content

Commit 2eaff5d

Browse files
razarmehrkulinseth
authored andcommitted
[MPS] Fix the uint8 type issue with View ops kernels (pytorch#95145)
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. Fixes pytorch#86954 Pull Request resolved: pytorch#95145 Approved by: https://github.com/kulinseth, https://github.com/DenisVieriu97
1 parent 7ebb309 commit 2eaff5d

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
@@ -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)