Skip to content

Commit 79927e2

Browse files
committed
[HIP] Implement num regs kernel query
1 parent cfe5912 commit 79927e2

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

source/adapters/hip/kernel.cpp

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -225,6 +225,12 @@ UR_APIEXPORT ur_result_t UR_APICALL urKernelGetInfo(ur_kernel_handle_t hKernel,
225225
return ReturnValue(hKernel->getProgram());
226226
case UR_KERNEL_INFO_ATTRIBUTES:
227227
return ReturnValue("");
228+
case UR_KERNEL_INFO_NUM_REGS: {
229+
int NumRegs = 0;
230+
UR_CHECK_ERROR(hipFuncGetAttribute(&NumRegs, HIP_FUNC_ATTRIBUTE_NUM_REGS,
231+
hKernel->get()));
232+
return ReturnValue(static_cast<uint32_t>(NumRegs));
233+
}
228234
default:
229235
break;
230236
}

test/conformance/kernel/kernel_adapter_hip.match

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,4 @@
11
urKernelGetGroupInfoWgSizeTest.CompileWorkGroupSize/*
2-
urKernelGetInfoTest.Success/*__UR_KERNEL_INFO_NUM_REGS
32
urKernelSetArgLocalTest.InvalidKernelArgumentIndex/*
43
urKernelSetArgMemObjTest.InvalidKernelArgumentIndex/*
54
urKernelSetArgPointerNegativeTest.InvalidKernelArgumentIndex/*

0 commit comments

Comments
 (0)