Fix hipMemoryAllocate test for single GPU

Change-Id: If121c18ab490ba125dc689ffc08a8839fd280c38


[ROCm/clr commit: 06ee0d3704]
此提交包含在:
Maneesh Gupta
2017-05-31 10:16:57 +05:30
父節點 2985fa3814
當前提交 0e4e17db27
+4 -3
查看文件
@@ -36,7 +36,6 @@ void multiGpuHostAlloc(int allocDevice)
int numDevices;
HIPCHECK(hipGetDeviceCount(&numDevices));
assert(numDevices > 1);
printf ("info: trying multiGpuHostAlloc with allocDevice=%d numDevices=%d\n", allocDevice, numDevices);
@@ -121,10 +120,12 @@ int main(int argc, char *argv[])
{
int numDevices;
HIPCHECK(hipGetDeviceCount(&numDevices));
assert(numDevices > 1);
multiGpuHostAlloc(0);
multiGpuHostAlloc(1);
if (numDevices > 1)
{
multiGpuHostAlloc(1);
}
}
passed();