From 5e32a3fcd27adfc57cee6edb41dc8baade3fda26 Mon Sep 17 00:00:00 2001
From: Vladana Stojiljkovic <Vladana.Stojiljkovic@amd.com>
Date: Tue, 6 Aug 2024 18:51:36 +0200
Subject: [PATCH] SWDEV-477597 - Implement tests for hipTexRefSetFlags

Change-Id: I0651882fe2eef1565a4fedfdd2fa4b23e4bb3f0a
---
 catch/unit/texture/CMakeLists.txt       |  1 +
 catch/unit/texture/hipTexRefSetFlags.cc | 75 +++++++++++++++++++++++++
 2 files changed, 76 insertions(+)
 create mode 100644 catch/unit/texture/hipTexRefSetFlags.cc

diff --git a/catch/unit/texture/CMakeLists.txt b/catch/unit/texture/CMakeLists.txt
index d857341211..4dc1e62f76 100644
--- a/catch/unit/texture/CMakeLists.txt
+++ b/catch/unit/texture/CMakeLists.txt
@@ -58,6 +58,7 @@ set(TEST_SRC
     hipTexRefSetMaxAnisotropy.cc
     hipTexRefGetMaxAnisotropy.cc
     hipUnbindTexture.cc
+    hipTexRefSetFlags.cc
 )
 
 # tests not for gfx90a+
diff --git a/catch/unit/texture/hipTexRefSetFlags.cc b/catch/unit/texture/hipTexRefSetFlags.cc
new file mode 100644
index 0000000000..93a725016b
--- /dev/null
+++ b/catch/unit/texture/hipTexRefSetFlags.cc
@@ -0,0 +1,75 @@
+/*
+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_hipTexRefSetFlags_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"));
+
+  SECTION("Null texture") {
+#if HT_AMD
+    HIP_CHECK_ERROR(hipTexRefSetFlags(nullptr, HIP_TRSF_READ_AS_INTEGER), hipErrorInvalidValue);
+#else
+    HIP_CHECK_ERROR(hipTexRefSetFlags(nullptr, HIP_TRSF_READ_AS_INTEGER),
+                    hipErrorInvalidResourceHandle);
+#endif
+  }
+
+  HIP_CHECK(hipModuleUnload(module));
+  HIP_CHECK(hipCtxDestroy(ctx));
+}
+
+TEST_CASE("Unit_hipTexRefSetFlags_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));
+
+  HIP_CHECK(hipModuleUnload(module));
+  HIP_CHECK(hipCtxDestroy(ctx));
+}
+
+#endif