Skip to content

Commit 4334b8b

Browse files
Context.create_module_ptx wraps PTX in an ObjectCode instance
1 parent 172e340 commit 4334b8b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

numba_cuda/numba/cuda/cudadrv/driver.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1436,7 +1436,7 @@ def create_module_ptx(self, ptx):
14361436
if isinstance(ptx, str):
14371437
ptx = ptx.encode('utf8')
14381438
if USE_NV_BINDING:
1439-
image = ptx
1439+
image = ObjectCode.from_ptx(ptx)
14401440
else:
14411441
image = c_char_p(ptx)
14421442
return self.create_module_image(image)

0 commit comments

Comments
 (0)