Fix wrong truncation of MPSShape from int64_t to int32_t (#58)

kulinseth · Jul 14, 2022 · 9c48392 · 9c48392
1 parent 65e542a
commit 9c48392
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/aten/src/ATen/native/mps/OperationUtils.mm b/aten/src/ATen/native/mps/OperationUtils.mm
@@ -192,7 +192,7 @@ MPSDataType getMPSScalarType(ScalarType scalar_type) {
  {
  NSInteger sz_i = (i < sz) ? t.size(i) : 1;
 
- NSNumber* number = [NSNumber numberWithInt:sz_i];
+ NSNumber* number = [NSNumber numberWithInteger:sz_i];
  numbers[i] = number;
  }
  return [NSArray arrayWithObjects:numbers count:sz_];
@@ -213,7 +213,7 @@ MPSDataType getMPSScalarType(ScalarType scalar_type) {
  {
  NSInteger sz_i = (i < sz) ? sizes[i] : 1;
 
- NSNumber* number = [NSNumber numberWithInt:sz_i];
+ NSNumber* number = [NSNumber numberWithInteger:sz_i];
  numbers[i] = number;
  }
  return [NSArray arrayWithObjects:numbers count:sz_];