From d58018ee1b362f7a6ae14b82d2e459a7856c481c Mon Sep 17 00:00:00 2001
From: Vladana Stojiljkovic <Vladana.Stojiljkovic@amd.com>
Date: Tue, 6 Aug 2024 11:56:09 +0200
Subject: [PATCH] SWDEV-476836 - Implement tests for hipTexRefGetAddressMode

Change-Id: Ia2003d0a7736145e745a55941ae6927100ba96a6
---
 catch/unit/texture/CMakeLists.txt             |   1 +
 catch/unit/texture/hipTexRefGetAddressMode.cc | 102 ++++++++++++++++++
 2 files changed, 103 insertions(+)
 create mode 100644 catch/unit/texture/hipTexRefGetAddressMode.cc

diff --git a/catch/unit/texture/CMakeLists.txt b/catch/unit/texture/CMakeLists.txt
index 981bfc71c8..c0decc57b4 100644
--- a/catch/unit/texture/CMakeLists.txt
+++ b/catch/unit/texture/CMakeLists.txt
@@ -62,6 +62,7 @@ set(TEST_SRC
     hipBindTextureToArray.cc
     hipTexRefGetFlags.cc
     hipTexRefSetAddressMode.cc
+    hipTexRefGetAddressMode.cc
 )
 
 # tests not for gfx90a+
diff --git a/catch/unit/texture/hipTexRefGetAddressMode.cc b/catch/unit/texture/hipTexRefGetAddressMode.cc
new file mode 100644
index 0000000000..093cfe025f
--- /dev/null
+++ b/catch/unit/texture/hipTexRefGetAddressMode.cc
@@ -0,0 +1,102 @@
+/*
+Copyright (c) 2024 Advanced Micro Devices, Inc. All rights reserved.
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+
+#pragma clang diagnostic ignored "-Wunused-parameter"
+#include <hip_test_common.hh>
+#include <hip/texture_types.h>
+
+#if defined(__HIP_PLATFORM_AMD__) || CUDA_VERSION < CUDA_12000
+
+TEST_CASE("Unit_hipTexRefGetAddressMode_Negative_Parameters") {
+  CHECK_IMAGE_SUPPORT
+
+  hipCtx_t ctx;
+  hipDevice_t device;
+
+  HIP_CHECK(hipGetDevice(&device));
+  HIP_CHECK(hipCtxCreate(&ctx, 0, device));
+
+  hipTexRef tex_ref = nullptr;
+  hipModule_t module = nullptr;
+  HIP_CHECK(hipModuleLoad(&module, "tex_ref_get_module.code"));
+  HIP_CHECK(hipModuleGetTexRef(&tex_ref, module, "tex"));
+
+  int dim = 0;
+
+#if HT_AMD
+  hipTextureAddressMode am;
+#else
+  HIPaddress_mode am;
+#endif
+
+  SECTION("Null texture") {
+#if HT_AMD
+    HIP_CHECK_ERROR(hipTexRefGetAddressMode(&am, nullptr, dim), hipErrorInvalidValue);
+#else
+    HIP_CHECK_ERROR(hipTexRefGetAddressMode(&am, nullptr, dim), hipErrorInvalidResourceHandle);
+#endif
+  }
+
+  SECTION("Null address mode") {
+    HIP_CHECK_ERROR(hipTexRefGetAddressMode(nullptr, tex_ref, dim), hipErrorInvalidValue);
+  }
+
+  SECTION("Invalid dimension") {
+    dim = GENERATE(-1, 3);
+    HIP_CHECK_ERROR(hipTexRefGetAddressMode(&am, tex_ref, dim), hipErrorInvalidValue);
+    HIP_CHECK_ERROR(hipTexRefGetAddressMode(&am, tex_ref, dim), hipErrorInvalidValue);
+  }
+
+  HIP_CHECK(hipModuleUnload(module));
+  HIP_CHECK(hipCtxDestroy(ctx));
+}
+
+TEST_CASE("Unit_hipTexRefGetAddressMode_Positive") {
+  CHECK_IMAGE_SUPPORT
+
+  hipCtx_t ctx;
+  hipDevice_t device;
+
+  HIP_CHECK(hipGetDevice(&device));
+  HIP_CHECK(hipCtxCreate(&ctx, 0, device));
+
+  hipTexRef tex_ref = nullptr;
+  hipModule_t module = nullptr;
+  HIP_CHECK(hipModuleLoad(&module, "tex_ref_get_module.code"));
+  HIP_CHECK(hipModuleGetTexRef(&tex_ref, module, "tex"));
+
+  int dim = 0;
+#if HT_AMD
+  hipTextureAddressMode am =
+      GENERATE(hipAddressModeWrap, hipAddressModeClamp, hipAddressModeMirror, hipAddressModeBorder);
+  hipTextureAddressMode out_am;
+#else
+  HIPaddress_mode am = GENERATE(HIP_TR_ADDRESS_MODE_WRAP, HIP_TR_ADDRESS_MODE_CLAMP);
+  HIPaddress_mode out_am;
+#endif
+
+  HIP_CHECK(hipTexRefSetAddressMode(tex_ref, dim, am));
+  HIP_CHECK(hipTexRefGetAddressMode(&out_am, tex_ref, dim));
+  REQUIRE(out_am == am);
+
+  HIP_CHECK(hipModuleUnload(module));
+  HIP_CHECK(hipCtxDestroy(ctx));
+}
+
+#endif