diff --git a/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h b/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h index bf53844646..05542a01b7 100644 --- a/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h +++ b/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h @@ -1709,6 +1709,15 @@ typedef cudaUserObject_t hipUserObject_t; #if CUDA_VERSION >= CUDA_12030 typedef cudaGraphEdgeData hipGraphEdgeData; #endif +typedef cudaLaunchConfig_t hipLaunchConfig_t; +typedef cudaLaunchAttribute hipLaunchAttribute; +typedef CUlaunchAttribute hipDrvLaunchAttribute; +typedef cudaKernel_t hipKernel_t; +typedef CUlaunchConfig HIP_LAUNCH_CONFIG; +typedef CUlaunchAttributeID hipDrvLaunchAttributeID; +typedef CUlaunchAttributeValue hipDrvLaunchAttributeValue; +#define hipLaunchAttributeCooperative cudaLaunchAttributeCooperative +#define hipDrvLaunchAttributeCooperative CU_LAUNCH_ATTRIBUTE_COOPERATIVE typedef enum cudaGraphNodeType hipGraphNodeType; #define hipGraphNodeTypeKernel cudaGraphNodeTypeKernel @@ -3405,6 +3414,11 @@ inline static hipError_t hipLaunchKernel(const void* function_address, dim3 numB cudaLaunchKernel(function_address, numBlocks, dimBlocks, args, sharedMemBytes, stream)); } +inline static hipError_t hipLaunchKernelExC(const hipLaunchConfig_t* config, const void* func, void** args) { + return hipCUDAErrorTohipError( + cudaLaunchKernelExC(config, func, args)); +} + inline static hipError_t hipModuleLaunchKernel(hipFunction_t f, unsigned int gridDimX, unsigned int gridDimY, unsigned int gridDimZ, unsigned int blockDimX, unsigned int blockDimY, @@ -3802,6 +3816,17 @@ inline static hipError_t hipLaunchCooperativeKernel(T f, dim3 gridDim, dim3 bloc cudaLaunchCooperativeKernel(reinterpret_cast(f), gridDim, blockDim, kernelParams, sharedMemBytes, stream)); } +inline static hipError_t hipDrvLaunchKernelEx(const HIP_LAUNCH_CONFIG* config, hipFunction_t f, void** params, void** extra) { + return hipCUResultTohipError( + cuLaunchKernelEx(config, f, params, extra)); +} + +template +inline static hipError_t hipLaunchKernelEx(const hipLaunchConfig_t* config, void (*kernel)(KernelArgs...), Params&&... args) { + return hipCUDAErrorTohipError( + cudaLaunchKernelEx(config, kernel, std::forward(args)...)); +} + inline static hipError_t hipTexObjectCreate(hipTextureObject_t* pTexObject, const HIP_RESOURCE_DESC* pResDesc, const HIP_TEXTURE_DESC* pTexDesc,