From 8d92d1ebd72a9bd4b0e72a3ca4fed4ae3d4dc8d9 Mon Sep 17 00:00:00 2001
From: Wilkin Chau <wing-ki.chauwilkin@amd.com>
Date: Thu, 28 Feb 2019 22:42:46 +0000
Subject: [PATCH] Fix hipMemset3D test

Calculate the allocated size based on the width, height and depth.
---
 src/hip_memory.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/hip_memory.cpp b/src/hip_memory.cpp
index ef5f80aee4..8e29bee68e 100644
--- a/src/hip_memory.cpp
+++ b/src/hip_memory.cpp
@@ -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();