adding masa's fix from apache#7669

Co-authored-by: Masahiro Masuda <masahi129@gmail.com>

python/tvm/topi/cuda/scan.py

@@ -16,6 +16,8 @@
 # under the License.
 # pylint: disable=invalid-name, too-many-locals, too-many-statements
 "Scan related operators"
+import logging
+
 import tvm
 from tvm import te
 from tvm.contrib.thrust import can_use_thrust, can_use_rocthrust
@@ -109,7 +111,8 @@ def exclusive_scan_ir(data, output, reduction=None, binop=tvm.tir.generic.add):
         # The following algorithm performs parallel exclusive scan
         # Up Sweep of exclusive scan
         lim = tvm.tir.generic.cast(
-            tvm.tir.ceil(tvm.tir.log2(tvm.tir.generic.cast(scan_axis_size, "float64"))), "int64"
+            tvm.tir.ceil(tvm.tir.log2(tvm.tir.generic.cast(scan_axis_size, size_cast_dtype))),
+            "int64",
         )
         with ib.for_range(0, lim, dtype="int64") as l2_width:
             width = 2 << l2_width

tests/python/unittest/test_target_codegen_spirv.py

@@ -103,6 +103,16 @@ def check_mod(mod, x_np, res_np):
 
     check_mod(mod, x_np, res_np)
 
+    # One 64 bit and one 32 bit constants
+    dtype = "int32"
+    x = relay.var("x", shape=(relay.Any(),), dtype=dtype)
+    mod = tvm.IRModule()
+    mod["main"] = relay.Function([x], relay.cumsum(x))
+    x_np = np.random.randint(0, high=10, size=(10,)).astype(dtype)
+    res_np = np.cumsum(x_np)
+
+    check_mod(mod, x_np, res_np)
+
 
 if __name__ == "__main__":
     test_bool_load()