From c1cc0e4ffe45790b97f56ae704189f4295a535e4 Mon Sep 17 00:00:00 2001
From: Vladana Stojiljkovic <Vladana.Stojiljkovic@amd.com>
Date: Wed, 7 Aug 2024 12:17:40 +0200
Subject: [PATCH] SWDEV-477981 - Implement tests for hipTexRefGetFlags

Change-Id: Iaa57185aa190ee44eb4ca54bc96bfac777ba0dd2
---
 catch/unit/texture/CMakeLists.txt       |  1 +
 catch/unit/texture/hipTexRefGetFlags.cc | 84 +++++++++++++++++++++++++
 2 files changed, 85 insertions(+)
 create mode 100644 catch/unit/texture/hipTexRefGetFlags.cc

diff --git a/catch/unit/texture/CMakeLists.txt b/catch/unit/texture/CMakeLists.txt
index d6a3471783..faa1368f87 100644
--- a/catch/unit/texture/CMakeLists.txt
+++ b/catch/unit/texture/CMakeLists.txt
@@ -60,6 +60,7 @@ set(TEST_SRC
     hipUnbindTexture.cc
     hipTexRefSetFlags.cc
     hipBindTextureToArray.cc
+    hipTexRefGetFlags.cc
 )
 
 # tests not for gfx90a+
diff --git a/catch/unit/texture/hipTexRefGetFlags.cc b/catch/unit/texture/hipTexRefGetFlags.cc
new file mode 100644
index 0000000000..18ff1898e0
--- /dev/null
+++ b/catch/unit/texture/hipTexRefGetFlags.cc
@@ -0,0 +1,84 @@
+/*
+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_hipTexRefGetFlags_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"));
+
+  unsigned int flags;
+
+  SECTION("Null texture") {
+#if HT_AMD
+    HIP_CHECK_ERROR(hipTexRefGetFlags(&flags, nullptr), hipErrorInvalidValue);
+#else
+    HIP_CHECK_ERROR(hipTexRefGetFlags(&flags, nullptr), hipErrorInvalidResourceHandle);
+#endif
+  }
+
+  SECTION("Null flags") {
+    HIP_CHECK_ERROR(hipTexRefGetFlags(nullptr, tex_ref), hipErrorInvalidValue);
+  }
+
+  HIP_CHECK(hipModuleUnload(module));
+  HIP_CHECK(hipCtxDestroy(ctx));
+}
+
+TEST_CASE("Unit_hipTexRefGetFlags_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"));
+
+  unsigned int flags =
+      GENERATE(HIP_TRSF_READ_AS_INTEGER, HIP_TRSF_NORMALIZED_COORDINATES, HIP_TRSF_SRGB);
+  HIP_CHECK(hipTexRefSetFlags(tex_ref, flags));
+
+  unsigned int out_flags;
+  HIP_CHECK(hipTexRefGetFlags(&out_flags, tex_ref));
+  REQUIRE(out_flags == flags);
+
+  HIP_CHECK(hipModuleUnload(module));
+  HIP_CHECK(hipCtxDestroy(ctx));
+}
+
+#endif