Skip to content

Commit ac06157

Browse files
committed
Fix HIP lexic compile
1 parent b4e1b6d commit ac06157

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tensorforge/common/vm/lexic/hip_lexic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ def get_launch_size(self, func_name, block, shmem):
4040
def set_shmem_size(self, func_name, shmem):
4141
return f"""static bool shmemsizeset = false;
4242
if (!shmemsizeset) {{
43-
hipFuncSetAttribute({func_name}, hipFuncAttributeMaxDynamicSharedMemorySize, {shmem});
43+
hipFuncSetAttribute(static_cast<const void*>(&{func_name}), hipFuncAttributeMaxDynamicSharedMemorySize, {shmem});
4444
CHECK_ERR;
4545
shmemsizeset = true;
4646
}}

0 commit comments

Comments
 (0)