AI/rocm-systems
2
0
Форкнуть 0
Код Задачи Запросы на слияние Действия Пакеты Проекты Релизы Вики Активность

In the hipMemset2D and hipMemset3D tests synchronize with the default stream after performing an async memset.

Просмотр исходного кода
Этот коммит содержится в:
Vladislav Sytchenko
2019-10-15 17:15:49 -04:00
родитель f402b6d01a
Коммит cc5abec092
2 изменённых файлов: 7 добавлений и 5 удалений
Скачать Patch файл Скачать Diff файл Показать все файлы Свернуть все файлы
hipamd/tests/src/runtimeApi/memory/hipMemset2D.cpp
+4 -3
Просмотреть файл
@@ -45,7 +45,7 @@ bool testhipMemset2D(int memsetval,int p_gpuDevice)
char *A_d; char *A_d;
char *A_h; char *A_h;
bool testResult = true; bool testResult = true;
HIPCHECK ( hipMemAllocPitch((hipDeviceptr_t*)&A_d, &pitch_A, width , numH,16) ); HIPCHECK(hipMemAllocPitch((hipDeviceptr_t*)&A_d, &pitch_A, width , numH,16));
A_h = (char*)malloc(sizeElements); A_h = (char*)malloc(sizeElements);
HIPASSERT(A_h != NULL); HIPASSERT(A_h != NULL);
for (size_t i=0; i<elements; i++) { for (size_t i=0; i<elements; i++) {
@@ -89,8 +89,9 @@ bool testhipMemset2DAsync(int memsetval,int p_gpuDevice)
} }
hipStream_t stream; hipStream_t stream;
HIPCHECK(hipStreamCreate(&stream)); HIPCHECK(hipStreamCreate(&stream));
HIPCHECK ( hipMemset2DAsync(A_d, pitch_A, memsetval, numW, numH, stream) ); HIPCHECK(hipMemset2DAsync(A_d, pitch_A, memsetval, numW, numH, stream) );
HIPCHECK ( hipMemcpy2D(A_h, width, A_d, pitch_A, numW, numH, hipMemcpyDeviceToHost)); HIPCHECK(hipStreamSynchronize(stream));
HIPCHECK(hipMemcpy2D(A_h, width, A_d, pitch_A, numW, numH, hipMemcpyDeviceToHost));
for (int i=0; i<elements; i++) { for (int i=0; i<elements; i++) {
if (A_h[i] != memsetval) { if (A_h[i] != memsetval) {
hipamd/tests/src/runtimeApi/memory/hipMemset3D.cpp
+3 -2
Просмотреть файл
@@ -53,7 +53,7 @@ bool testhipMemset3D(int memsetval,int p_gpuDevice)
for (size_t i=0; i<elements; i++) { for (size_t i=0; i<elements; i++) {
A_h[i] = 1; A_h[i] = 1;
} }
HIPCHECK ( hipMemset3D( devPitchedPtr, memsetval, extent) ); HIPCHECK(hipMemset3D( devPitchedPtr, memsetval, extent));
hipMemcpy3DParms myparms = {0}; hipMemcpy3DParms myparms = {0};
myparms.srcPos = make_hipPos(0,0,0); myparms.srcPos = make_hipPos(0,0,0);
myparms.dstPos = make_hipPos(0,0,0); myparms.dstPos = make_hipPos(0,0,0);
@@ -103,7 +103,8 @@ bool testhipMemset3DAsync(int memsetval,int p_gpuDevice)
} }
hipStream_t stream; hipStream_t stream;
HIPCHECK(hipStreamCreate(&stream)); HIPCHECK(hipStreamCreate(&stream));
HIPCHECK ( hipMemset3DAsync( devPitchedPtr, memsetval, extent, stream) ); HIPCHECK(hipMemset3DAsync(devPitchedPtr, memsetval, extent, stream));
HIPCHECK(hipStreamSynchronize(stream));
hipMemcpy3DParms myparms = {0}; hipMemcpy3DParms myparms = {0};
myparms.srcPos = make_hipPos(0,0,0); myparms.srcPos = make_hipPos(0,0,0);
myparms.dstPos = make_hipPos(0,0,0); myparms.dstPos = make_hipPos(0,0,0);