Fix hipMemset3D test

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


[ROCm/hip commit: 8d92d1ebd7]
This commit is contained in:
Wilkin Chau
2019-02-28 22:42:46 +00:00
parent 2606b86405
commit 4a0d68ba3f
+1 -1
Vedi File
@@ -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();