Skip to content

Commit f70c06a

Browse files
author
Diptorup Deb
committed
Pass types.int32 as a 64-bit value as well.
1 parent 732b654 commit f70c06a

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

numba_dpex/dpctl_iface/_helpers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,9 @@ def numba_type_to_dpctl_typenum(context, ty):
1717
from dpctl.enum_types import kernel_arg_type as kargty
1818

1919
if ty == types.int32 or isinstance(ty, types.scalars.IntegerLiteral):
20-
return context.get_constant(types.int32, kargty.dpctl_int32.value)
20+
return context.get_constant(types.int32, kargty.dpctl_int64.value)
2121
elif ty == types.uint32:
22-
return context.get_constant(types.int32, kargty.dpctl_uint32.value)
22+
return context.get_constant(types.int32, kargty.dpctl_uint64.value)
2323
elif ty == types.boolean:
2424
return context.get_constant(types.int32, kargty.dpctl_uint32.value)
2525
elif ty == types.int64:

0 commit comments

Comments
 (0)