Fix hipMemset3D test

Calculate the allocated size based on the width, height and depth.


[ROCm/hip commit: 8d92d1ebd7]
Este commit está contenido en:
Wilkin Chau
2019-02-28 22:42:46 +00:00
padre 2606b86405
commit 4a0d68ba3f
+1 -1
Ver fichero
@@ -350,7 +350,7 @@ hipError_t ihipMallocPitch(void** ptr, size_t* pitch, size_t width, size_t heigh
}
// hardcoded 128 bytes
*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();