AI/rocm-systems
2
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

SWDEV-553962 - Add p2p check to multidevice virtual memory tests (#898)

Browse Source
This commit is contained in:
marandje authored and GitHub committed 2025-09-11 01:15:18 +02:00
1 parent 8cb6278f32
commit ea68344f2f
1 file changed
+6 -3
Download Patch File Download Diff File Expand all files Collapse all files
projects/hip-tests/catch/unit/virtualMemoryManagement/hipMemMap.cc
+6 -3
View File
@@ -276,6 +276,7 @@ TEST_CASE("Unit_hipMemMap_PhysicalMemory_Map2MultVMMs") {
* - HIP_VERSION >= 6.1 * - HIP_VERSION >= 6.1
*/ */
TEST_CASE("Unit_hipMemMap_PhysicalMemoryReuse_MultiDev") { TEST_CASE("Unit_hipMemMap_PhysicalMemoryReuse_MultiDev") {
CHECK_P2P_SUPPORT
int devicecount = 0; int devicecount = 0;
HIP_CHECK(hipGetDeviceCount(&devicecount)); HIP_CHECK(hipGetDeviceCount(&devicecount));
if (devicecount < 2) { if (devicecount < 2) {
@@ -423,6 +424,7 @@ TEST_CASE("Unit_hipMemMap_VMMMemoryReuse_SingleGPU") {
* - HIP_VERSION >= 6.1 * - HIP_VERSION >= 6.1
*/ */
TEST_CASE("Unit_hipMemMap_VMMMemoryReuse_MultiGPU") { TEST_CASE("Unit_hipMemMap_VMMMemoryReuse_MultiGPU") {
CHECK_P2P_SUPPORT
int deviceId = 0, devicecount = 0; int deviceId = 0, devicecount = 0;
HIP_CHECK(hipGetDeviceCount(&devicecount)); HIP_CHECK(hipGetDeviceCount(&devicecount));
if (devicecount < 2) { if (devicecount < 2) {
@@ -432,6 +434,7 @@ TEST_CASE("Unit_hipMemMap_VMMMemoryReuse_MultiGPU") {
size_t granularity = 0; size_t granularity = 0;
size_t buffer_size = N * sizeof(int); size_t buffer_size = N * sizeof(int);
hipDevice_t device; hipDevice_t device;
HIP_CHECK(hipSetDevice(0));
HIP_CHECK(hipDeviceGet(&device, deviceId)); HIP_CHECK(hipDeviceGet(&device, deviceId));
checkVMMSupported(device); checkVMMSupported(device);
hipMemAllocationProp prop{}; hipMemAllocationProp prop{};
@@ -451,9 +454,9 @@ TEST_CASE("Unit_hipMemMap_VMMMemoryReuse_MultiGPU") {
} }
// Allocate a physical memory chunk // Allocate a physical memory chunk
for (int dev = 0; dev < devicecount; dev++) { for (int dev = 0; dev < devicecount; dev++) {
hipDevice_t device; hipDevice_t dev_handle;
HIP_CHECK(hipDeviceGet(&device, dev)); HIP_CHECK(hipDeviceGet(&dev_handle, dev));
prop.location.id = device; prop.location.id = dev_handle;
HIP_CHECK(hipMemCreate(&handle[dev], size_mem, &prop, 0)); HIP_CHECK(hipMemCreate(&handle[dev], size_mem, &prop, 0));
} }
// Allocate devicecount virtual address ranges // Allocate devicecount virtual address ranges