Skip to content

Commit a2edb3f

Browse files
committed
[lang] Fix pointer dtype bitwidth
Signed-off-by: Greg Bonik <gbonik@nvidia.com>
1 parent 454a5ce commit a2edb3f

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/cuda/tile/_datatype.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -657,7 +657,7 @@ def _get_pointer_dtype(pointee_dtype: DType | None, memory_space: MemorySpace) -
657657
pass
658658

659659
match memory_space:
660-
case MemorySpace.SHARED:
660+
case MemorySpace.SHARED | MemorySpace.TENSOR | MemorySpace.SHARED_CLUSTER:
661661
bitwidth = 32
662662
case _:
663663
bitwidth = 64

0 commit comments

Comments
 (0)