Merge pull request #2355 from npmiller/hip-num-regs

[HIP] Implement num regs kernel query

Author: kbenzie

source/adapters/hip/kernel.cpp

@@ -226,6 +226,12 @@ UR_APIEXPORT ur_result_t UR_APICALL urKernelGetInfo(ur_kernel_handle_t hKernel,
     return ReturnValue(hKernel->getProgram());
   case UR_KERNEL_INFO_ATTRIBUTES:
     return ReturnValue("");
+  case UR_KERNEL_INFO_NUM_REGS: {
+    int NumRegs = 0;
+    UR_CHECK_ERROR(hipFuncGetAttribute(&NumRegs, HIP_FUNC_ATTRIBUTE_NUM_REGS,
+                                       hKernel->get()));
+    return ReturnValue(static_cast<uint32_t>(NumRegs));
+  }
   default:
     break;
   }

test/conformance/kernel/kernel_adapter_hip.match

@@ -1,5 +1,4 @@
 urKernelGetGroupInfoWgSizeTest.CompileWorkGroupSize/*
-urKernelGetInfoTest.Success/*__UR_KERNEL_INFO_NUM_REGS
 urKernelSetArgLocalTest.InvalidKernelArgumentIndex/*
 urKernelSetArgMemObjTest.InvalidKernelArgumentIndex/*
 urKernelSetArgPointerNegativeTest.InvalidKernelArgumentIndex/*