Small fix

NVIDIA · Feb 6, 2025 · e8bc8fc · e8bc8fc
1 parent 9e6562e
commit e8bc8fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/cuda_parallel/cuda/parallel/experimental/algorithms/reduce.py b/python/cuda_parallel/cuda/parallel/experimental/algorithms/reduce.py
@@ -110,7 +110,7 @@ def __call__(
         else:
             self.d_out_cccl.state = d_out.state
 
-        if self.h_init_cccl.type.type == cccl.TypeEnum.STORAGE:
+        if self.h_init_cccl.type.type.value == cccl.TypeEnum.STORAGE:
             self.h_init_cccl.state = h_init._data.__array_interface__["data"][0]  # type: ignore
         else:
             self.h_init_cccl.state = h_init.__array_interface__["data"][0]