AI/rocm-systems
2
0
Förgrening 0
Kod Ärenden Pull-förfrågningar Actions Packages Projekt Släpp Wiki Aktiviteter

SWDEV-1 - Add missing checks in hipMemcpy_MultiThread.cc

Bläddra i källkod
This commit is contained in:
Maneesh Gupta
2022-10-11 08:53:04 +05:30
committad av GitHub
förälder dcf32d639a
incheckning 79eeb9b806
1 ändrade filer med 4 tillägg och 4 borttagningar
Download Patch File Download Diff File Expand all files Collapse all files
catch/unit/memory/hipMemcpy_MultiThread.cc
+4 -4
Visa fil
@@ -105,7 +105,7 @@ void memcpyTests<T>::Memcpy_And_verify(bool *ret_val) {
for (int i = 0; i < Available_Gpus; ++i) {
for (int j = i+1; j < Available_Gpus; ++j) {
canAccessPeer = 0;
hipDeviceCanAccessPeer(&canAccessPeer, i, j);
HIPCHECK(hipDeviceCanAccessPeer(&canAccessPeer, i, j));
if (canAccessPeer) {
HIPCHECK(hipMemcpy(A_d[j], A_d[i], NUM_ELM * sizeof(T),
hipMemcpyDefault));
@@ -155,7 +155,7 @@ void memcpyTests<T>::Memcpy_And_verify(bool *ret_val) {
int canAccessPeer = 0;
for (int i = 0; i < Available_Gpus; ++i) {
for (int j = i+1; j < Available_Gpus; ++j) {
hipDeviceCanAccessPeer(&canAccessPeer, i, j);
HIPCHECK(hipDeviceCanAccessPeer(&canAccessPeer, i, j));
if (canAccessPeer) {
HIPCHECK(hipMemcpyHtoD(hipDeviceptr_t(A_d[i]),
A_h, NUM_ELM * sizeof(T)));
@@ -198,7 +198,7 @@ void memcpyTests<T>::Memcpy_And_verify(bool *ret_val) {
for (int i = 0; i < Available_Gpus; ++i) {
for (int j = i+1; j < Available_Gpus; ++j) {
canAccessPeer = 0;
hipDeviceCanAccessPeer(&canAccessPeer, i, j);
HIPCHECK(hipDeviceCanAccessPeer(&canAccessPeer, i, j));
if (canAccessPeer) {
HIPCHECK(hipMemcpyAsync(A_d[j], A_d[i],
NUM_ELM * sizeof(T),
@@ -252,7 +252,7 @@ void memcpyTests<T>::Memcpy_And_verify(bool *ret_val) {
for (int i = 0; i < Available_Gpus; ++i) {
for (int j = i+1; j < Available_Gpus; ++j) {
canAccessPeer = 0;
hipDeviceCanAccessPeer(&canAccessPeer, i, j);
HIPCHECK(hipDeviceCanAccessPeer(&canAccessPeer, i, j));
if (canAccessPeer) {
HIPCHECK(hipSetDevice(j));
HIPCHECK(hipMemcpyDtoDAsync(hipDeviceptr_t(A_d[j]),