fixed bug on specification of dtype parameter

Owen Elliott · Owen Elliott · commit 73d9dccad9b8 · 2022-12-18T23:59:55.000+11:00
diff --git a/aten/src/ATen/native/mps/operations/RangeFactories.mm b/aten/src/ATen/native/mps/operations/RangeFactories.mm
@@ -132,7 +132,15 @@
     auto xend = end.to<accscalar_t>();
     auto xstep = step.to<accscalar_t>();
     
-    double size_d = ((xend - xstart) / xstep) + 1;
+    // double size_d = ((xend - xstart) / xstep) + 1;
+    double size_d;
+    if (std::is_same<scalar_t, int64_t>::value) {
+      size_d = static_cast<double>(end.to<accscalar_t>() - start.to<accscalar_t>())
+                / step.to<accscalar_t>() + 1;
+    } else {
+      size_d = static_cast<double>(end.to<double>() - start.to<double>())
+                / step.to<double>() + 1;
+    }
 
     TORCH_CHECK(xstep > 0 || xstep < 0, "step must be nonzero");
     TORCH_CHECK(std::isfinite(static_cast<double>(xstart)) &&
