From e191c5876e5c29653fe0e8abd4510f355c267e55 Mon Sep 17 00:00:00 2001
From: "Jobczyk, Lukasz" <lukasz.jobczyk@intel.com>
Date: Wed, 6 Feb 2019 13:39:23 +0100
Subject: [PATCH] Adding buffer tests that check an L3 setting

Change-Id: Ib7759fc7430c931f6f24337d852a8644abbb199e
Signed-off-by: Jobczyk, Lukasz <lukasz.jobczyk@intel.com>
---
 Jenkinsfile                         |  2 +-
 unit_tests/mem_obj/buffer_tests.cpp | 70 +++++++++++++++++++++++++++++
 2 files changed, 71 insertions(+), 1 deletion(-)

diff --git a/Jenkinsfile b/Jenkinsfile
index e935be0b1a..cd4c31c6f5 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -1,5 +1,5 @@
 #!groovy
 dependenciesRevision='88b9b8081c963d105ae1815384e62d225a92ca51-1208'
 strategy='EQUAL'
-allowedCD=270
+allowedCD=268
 allowedF=4
diff --git a/unit_tests/mem_obj/buffer_tests.cpp b/unit_tests/mem_obj/buffer_tests.cpp
index 3a44ad33bb..23993959bf 100644
--- a/unit_tests/mem_obj/buffer_tests.cpp
+++ b/unit_tests/mem_obj/buffer_tests.cpp
@@ -6,6 +6,7 @@
  */
 
 #include "gmock/gmock.h"
+#include "runtime/command_queue/command_queue_hw.h"
 #include "runtime/gmm_helper/gmm.h"
 #include "runtime/gmm_helper/gmm_helper.h"
 #include "runtime/gmm_helper/resource_info.h"
@@ -1421,6 +1422,75 @@ HWTEST_F(BufferSetSurfaceTests, givenBufferThatIsMisalignedWhenSurfaceStateIsBei
     EXPECT_EQ(0u, surfaceState.getMemoryObjectControlState());
 }
 
+class BufferL3CacheTests : public ::testing::TestWithParam<uint64_t> {
+  public:
+    void SetUp() override {
+        ctx.getDevice(0)->getExecutionEnvironment()->getGmmHelper()->setSimplifiedMocsTableUsage(true);
+        hostPtr = reinterpret_cast<void *>(GetParam());
+    }
+    MockContext ctx;
+    const size_t region[3] = {3, 3, 1};
+    const size_t origin[3] = {0, 0, 0};
+
+    void *hostPtr;
+};
+
+HWTEST_P(BufferL3CacheTests, givenMisalignedAndAlignedBufferWhenClEnqueueWriteImageThenL3CacheIsOn) {
+    using RENDER_SURFACE_STATE = typename FamilyType::RENDER_SURFACE_STATE;
+
+    CommandQueueHw<FamilyType> cmdQ(&ctx, ctx.getDevice(0), nullptr);
+    auto surfaceState = reinterpret_cast<RENDER_SURFACE_STATE *>(cmdQ.getCommandStreamReceiver().getIndirectHeap(IndirectHeap::Type::SURFACE_STATE, 0).getSpace(0));
+
+    cl_image_format imageFormat;
+    cl_image_desc imageDesc;
+    imageFormat.image_channel_order = CL_RGBA;
+    imageFormat.image_channel_data_type = CL_UNORM_INT8;
+    imageDesc.image_type = CL_MEM_OBJECT_IMAGE2D;
+    imageDesc.image_width = 3;
+    imageDesc.image_height = 3;
+    imageDesc.image_depth = 1;
+    imageDesc.image_array_size = 1;
+    imageDesc.image_row_pitch = 0;
+    imageDesc.image_slice_pitch = 0;
+    imageDesc.num_mip_levels = 0;
+    imageDesc.num_samples = 0;
+    imageDesc.mem_object = nullptr;
+    auto image = clCreateImage(&ctx, CL_MEM_READ_WRITE, &imageFormat, &imageDesc, nullptr, nullptr);
+
+    clEnqueueWriteImage(&cmdQ, image, false, origin, region, 0, 0, hostPtr, 0, nullptr, nullptr);
+
+    auto expect = ctx.getDevice(0)->getExecutionEnvironment()->getGmmHelper()->getMOCS(GMM_RESOURCE_USAGE_OCL_BUFFER);
+    EXPECT_NE(NULL, surfaceState->getMemoryObjectControlState());
+    EXPECT_EQ(expect, surfaceState->getMemoryObjectControlState());
+
+    clReleaseMemObject(image);
+}
+
+HWTEST_P(BufferL3CacheTests, givenMisalignedAndAlignedBufferWhenClEnqueueWriteBufferRectThenL3CacheIsOn) {
+    using RENDER_SURFACE_STATE = typename FamilyType::RENDER_SURFACE_STATE;
+
+    CommandQueueHw<FamilyType> cmdQ(&ctx, ctx.getDevice(0), nullptr);
+    auto surfaceState = reinterpret_cast<RENDER_SURFACE_STATE *>(cmdQ.getCommandStreamReceiver().getIndirectHeap(IndirectHeap::Type::SURFACE_STATE, 0).getSpace(0));
+    auto buffer = clCreateBuffer(&ctx, CL_MEM_READ_WRITE, 36, nullptr, nullptr);
+
+    clEnqueueWriteBufferRect(&cmdQ, buffer, false, origin, origin, region, 0, 0, 0, 0, hostPtr, 0, nullptr, nullptr);
+
+    auto expect = ctx.getDevice(0)->getExecutionEnvironment()->getGmmHelper()->getMOCS(GMM_RESOURCE_USAGE_OCL_BUFFER);
+    EXPECT_NE(NULL, surfaceState->getMemoryObjectControlState());
+    EXPECT_EQ(expect, surfaceState->getMemoryObjectControlState());
+
+    clReleaseMemObject(buffer);
+}
+
+static uint64_t pointers[] = {
+    0x1005,
+    0x2000};
+
+INSTANTIATE_TEST_CASE_P(
+    pointers,
+    BufferL3CacheTests,
+    testing::ValuesIn(pointers));
+
 struct BufferUnmapTest : public DeviceFixture, public ::testing::Test {
     void SetUp() override {
         DeviceFixture::SetUp();