Update

pytorch · Jul 21, 2023 · 06ef382 · 06ef382
1 parent defae21
commit 06ef382
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch/_refs/__init__.py b/torch/_refs/__init__.py
@@ -4639,12 +4639,12 @@ def logspace(
     requires_grad: bool = False,
 ) -> TensorLikeType:
     # NB: NumPy doesn't have this cast
-    if prims.utils.is_integer_dtype(dtype):
+    if prims.utils.is_integer_dtype(dtype or torch.get_default_dtype()):
         if isinstance(start, FloatLike):
             start = sym_int(start)
         if isinstance(end, FloatLike):
             end = sym_int(end)
-    
+
     if py_any(isinstance(arg, complex) for arg in (start, end, steps)):
         default_complex_dtype = utils.corresponding_complex_dtype(
             torch.get_default_dtype()