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 a4e3fa4f1b
commit 99540373cf
+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();