From 98ef2f64b55a11c2d2ba8352bfc7b76a87fac701 Mon Sep 17 00:00:00 2001 From: "Dittakavi, Satyanvesh" Date: Tue, 12 Aug 2025 16:30:30 +0530 Subject: [PATCH] SWDEV-545947 - Add Nvidia mapping for hipSetValidDevices (#29) --- hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h | 4 ++++ 1 file changed, 4 insertions(+) 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 06ba5cf7f9..ee4ed49e41 100644 --- a/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h +++ b/hipnv/include/hip/nvidia_detail/nvidia_hip_runtime_api.h @@ -2097,6 +2097,10 @@ inline static hipError_t hipSetDevice(int device) { return hipCUDAErrorTohipError(cudaSetDevice(device)); } +inline static hipError_t hipSetValidDevices(int* device_arr, int len) { + return hipCUDAErrorTohipError(cudaSetValidDevices(device_arr, len)); +} + inline static hipError_t hipChooseDevice(int* device, const hipDeviceProp_t* prop) { if (prop == NULL) {