Fix hipMemset3D test

Calculate the allocated size based on the width, height and depth.
This commit is contained in:
Wilkin Chau
2019-02-28 22:42:46 +00:00
parent e8ca9cbf95
commit 8d92d1ebd7
+1 -1
View 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();