Skip to content

Commit 1c9e7aa

Browse files
committed
perf: avoid extra dtype call
1 parent 848dfcc commit 1c9e7aa

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

numba_cuda/numba/cuda/api.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -39,10 +39,9 @@ def from_cuda_array_interface(desc, owner=None, sync=True):
3939

4040
shape = desc["shape"]
4141
strides = desc.get("strides")
42-
dtype = np.dtype(desc["typestr"])
4342

4443
shape, strides, dtype = prepare_shape_strides_dtype(
45-
shape, strides, dtype, order="C"
44+
shape, strides, desc["typestr"], order="C"
4645
)
4746
size = driver.memory_size_from_info(shape, strides, dtype.itemsize)
4847

0 commit comments

Comments
 (0)