Compilation

intel · Jul 21, 2024 · 0e0b170 · 0e0b170
1 parent d4d1160
commit 0e0b170
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ATen/native/xpu/RangeFactories.cpp b/src/ATen/native/xpu/RangeFactories.cpp
@@ -19,7 +19,7 @@ Tensor& XPUNativeFunctions::arange_out(
   AT_DISPATCH_ALL_TYPES_AND2(
       at::ScalarType::Half,
       at::ScalarType::BFloat16,
-      result.scalar_type(),
+      out.scalar_type(),
       "arange_xpu_preprocess",
       [&]() {
         using accscalar_t = at::acc_type<scalar_t, true>;