Fix hipMemset3D test

Calculate the allocated size based on the width, height and depth.
Esse commit está contido em:
Wilkin Chau
2019-02-28 22:42:46 +00:00
commit 8d92d1ebd7
+1 -1
Ver Arquivo
@@ -350,7 +350,7 @@ hipError_t ihipMallocPitch(void** ptr, size_t* pitch, size_t width, size_t heigh
} }
// hardcoded 128 bytes // hardcoded 128 bytes
*pitch = ((((int)width - 1) / 128) + 1) * 128; *pitch = ((((int)width - 1) / 128) + 1) * 128;
const size_t sizeBytes = (*pitch) * height; const size_t sizeBytes = (*pitch) * height * ((depth==0) ? 1 : depth);
auto ctx = ihipGetTlsDefaultCtx(); auto ctx = ihipGetTlsDefaultCtx();