/*
 * Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
 *
 * SPDX-License-Identifier: MIT
 */

#include "atomicExch_common.hh"
#include "atomicExch_system_negative_kernels_rtc.hh"

/**
 * @addtogroup atomicExch_system atomicExch_system
 * @{
 * @ingroup AtomicsTest
 */

// Helper function to run atomicExch_system tests for peer GPUs
template <typename TestType> static void runAtomicExchSystemPeerGPUsTest() {
  int warp_size = 0;
  HIP_CHECK(hipDeviceGetAttribute(&warp_size, hipDeviceAttributeWarpSize, 0));
  const auto cache_line_size = 128u;

  SECTION("Same address") {
    AtomicExchMultipleDeviceMultipleKernelAndHostTest<TestType>(2, 2, 1, sizeof(TestType));
  }

  SECTION("Adjacent addresses") {
    AtomicExchMultipleDeviceMultipleKernelAndHostTest<TestType>(2, 2, warp_size,
                                                                sizeof(TestType));
  }

  SECTION("Scattered addresses") {
    AtomicExchMultipleDeviceMultipleKernelAndHostTest<TestType>(2, 2, warp_size, cache_line_size);
  }
}

// Helper function to run atomicExch_system tests for host and GPU
template <typename TestType> static void runAtomicExchSystemHostAndGPUTest() {
  int warp_size = 0;
  HIP_CHECK(hipDeviceGetAttribute(&warp_size, hipDeviceAttributeWarpSize, 0));
  const auto cache_line_size = 128u;

  SECTION("Same address") {
    AtomicExchMultipleDeviceMultipleKernelAndHostTest<TestType>(1, 1, 1, sizeof(TestType), 4);
  }

  SECTION("Adjacent addresses") {
    AtomicExchMultipleDeviceMultipleKernelAndHostTest<TestType>(1, 1, warp_size, sizeof(TestType),
                                                                4);
  }

  SECTION("Scattered addresses") {
    AtomicExchMultipleDeviceMultipleKernelAndHostTest<TestType>(1, 1, warp_size, cache_line_size,
                                                                4);
  }
}

// Helper function to run atomicExch_system tests for host and peer GPUs
template <typename TestType> static void runAtomicExchSystemHostAndPeerGPUsTest() {
  int warp_size = 0;
  HIP_CHECK(hipDeviceGetAttribute(&warp_size, hipDeviceAttributeWarpSize, 0));
  const auto cache_line_size = 128u;

  SECTION("Same address") {
    AtomicExchMultipleDeviceMultipleKernelAndHostTest<TestType>(2, 2, 1, sizeof(TestType), 4);
  }

  SECTION("Adjacent addresses") {
    AtomicExchMultipleDeviceMultipleKernelAndHostTest<TestType>(2, 2, warp_size, sizeof(TestType),
                                                                4);
  }

  SECTION("Scattered addresses") {
    AtomicExchMultipleDeviceMultipleKernelAndHostTest<TestType>(2, 2, warp_size, cache_line_size,
                                                                4);
  }
}

/**
 * Test Description
 * ------------------------
 *    - Executes a kernel two times concurrently on two devices wherein all threads will perform
 * an atomic exchange into a runtime determined memory location. Each thread will exchange its own
 * grid wide linear index + offset into the memory location, storing the return value into a
 * separate output array slot corresponding to it. Once complete, the union of output array and
 * exchange memory is validated to contain all values in the range [0, number_of_threads +
 * number_of_exchange_memory_slots). Several memory access patterns are tested:
 *      -# All threads exchange to a single memory location
 *      -# Each thread exchanges into an array containing warp_size elements, using tid % warp_size
 * for indexing
 *      -# Same as the above, but the exchange elements are spread out by L1 cache line size bytes.
 *
 *    - The test is run for:
 *      - All overloads of atomicExch_system
 *      - hipMallocManaged, hipHostMalloc and hipHostRegister allocated exchange memory
 *      - Several grid and block dimension combinations
 * Test source
 * ------------------------
 *    - unit/atomics/atomicExch_system.cc
 * Test requirements
 * ------------------------
 *    - HIP_VERSION >= 5.2
 */
HIP_TEST_CASE(Unit_atomicExch_system_Positive_Peer_GPUs) {
  SECTION("int") { runAtomicExchSystemPeerGPUsTest<int>(); }
  SECTION("unsigned int") { runAtomicExchSystemPeerGPUsTest<unsigned int>(); }
#ifndef HT_NVIDIA
  SECTION("unsigned long") { runAtomicExchSystemPeerGPUsTest<unsigned long>(); }
#endif
  SECTION("unsigned long long") { runAtomicExchSystemPeerGPUsTest<unsigned long long>(); }
  SECTION("float") { runAtomicExchSystemPeerGPUsTest<float>(); }
#ifndef HT_NVIDIA
  SECTION("double") { runAtomicExchSystemPeerGPUsTest<double>(); }
#endif
}

/**
 * Test Description
 * ------------------------
 *    - Executes a kernel on a single device wherein all threads will perform an atomic exchange
 * into a runtime determined memory location. Each thread will exchange its own grid wide linear
 * index + offset into the memory location, storing the return value into a separate output array
 * slot corresponding to it. While the kernel is running, the host performs atomic exchanges, in 4
 * threads, into the same memory location(s). Once complete, the union of output array, exchange
 * memory, and host output is validated to contain all values in the range [0, number_of_threads +
 * number_of_exchange_memory_slots + number_of_host_iterations). Several memory access patterns are
 * tested:
 *      -# All threads exchange to a single memory location
 *      -# Each thread exchanges into an array containing warp_size elements, using tid % warp_size
 * for indexing
 *      -# Same as the above, but the exchange elements are spread out by L1 cache line size bytes.
 *
 *    - The test is run for:
 *      - All overloads of atomicExch_system
 *      - hipMallocManaged, hipHostMalloc and hipHostRegister allocated exchange memory
 *      - Several grid and block dimension combinations
 * Test source
 * ------------------------
 *    - unit/atomics/atomicExch_system.cc
 * Test requirements
 * ------------------------
 *    - HIP_VERSION >= 5.2
 */
HIP_TEST_CASE(Unit_atomicExch_system_Positive_Host_And_GPU) {
  SECTION("int") { runAtomicExchSystemHostAndGPUTest<int>(); }
  SECTION("unsigned int") { runAtomicExchSystemHostAndGPUTest<unsigned int>(); }
#ifndef HT_NVIDIA
  SECTION("unsigned long") { runAtomicExchSystemHostAndGPUTest<unsigned long>(); }
#endif
  SECTION("unsigned long long") { runAtomicExchSystemHostAndGPUTest<unsigned long long>(); }
  SECTION("float") { runAtomicExchSystemHostAndGPUTest<float>(); }
#ifndef HT_NVIDIA
  SECTION("double") { runAtomicExchSystemHostAndGPUTest<double>(); }
#endif
}

/**
 * Test Description
 * ------------------------
 *    - Executes a kernel two times concurrently on two devices wherein all threads will perform
 * an atomic exchange into a runtime determined memory location. Each thread will exchange its own
 * grid wide linear index + offset into the memory location, storing the return value into a
 * separate output array slot corresponding to it. While the kernels are running, the
 * host performs atomic exchanges, in 4 threads, into the same memory location(s). Once complete,
 * the union of output array, exchange memory, and host output is validated to contain all values in
 * the range [0, number_of_threads + number_of_exchange_memory_slots + number_of_host_iterations).
 * Several memory access patterns are tested:
 *      -# All threads exchange to a single memory location
 *      -# Each thread exchanges into an array containing warp_size elements, using tid % warp_size
 * for indexing
 *      -# Same as the above, but the exchange elements are spread out by L1 cache line size bytes.
 *
 *    - The test is run for:
 *      - All overloads of atomicExch_system
 *      - hipMallocManaged, hipHostMalloc and hipHostRegister allocated exchange memory
 *      - Several grid and block dimension combinations
 * Test source
 * ------------------------
 *    - unit/atomics/atomicExch_system.cc
 * Test requirements
 * ------------------------
 *    - HIP_VERSION >= 5.2
 */
HIP_TEST_CASE(Unit_atomicExch_system_Positive_Host_And_Peer_GPUs) {
  SECTION("int") { runAtomicExchSystemHostAndPeerGPUsTest<int>(); }
  SECTION("unsigned int") { runAtomicExchSystemHostAndPeerGPUsTest<unsigned int>(); }
#ifndef HT_NVIDIA
  SECTION("unsigned long") { runAtomicExchSystemHostAndPeerGPUsTest<unsigned long>(); }
#endif
  SECTION("unsigned long long") { runAtomicExchSystemHostAndPeerGPUsTest<unsigned long long>(); }
  SECTION("float") { runAtomicExchSystemHostAndPeerGPUsTest<float>(); }
#ifndef HT_NVIDIA
  SECTION("double") { runAtomicExchSystemHostAndPeerGPUsTest<double>(); }
#endif
}

/**
 * Test Description
 * ------------------------
 *    - RTCs kernels that pass combinations of arguments of invalid types for all overloads of
 * atomicExch_system
 * Test source
 * ------------------------
 *    - unit/atomics/atomicExch_system.cc
 * Test requirements
 * ------------------------
 *    - HIP_VERSION >= 5.2
 */
HIP_TEST_CASE(Unit_atomicExch_system_Negative_Parameters_RTC) {
  hiprtcProgram program{};

  const auto program_source =
      GENERATE(kAtomicExchSystemInt, kAtomicExchSystemUnsignedInt, kAtomicExchSystemULL,
               kAtomicExchSystemFloat, kAtomicExchSystemDouble);
  HIPRTC_CHECK(
      hiprtcCreateProgram(&program, program_source, "atomicExch_negative.cc", 0, nullptr, nullptr));
  hiprtcResult result{hiprtcCompileProgram(program, 0, nullptr)};

  // Get the compile log and count compiler error messages
  size_t log_size{};
  HIPRTC_CHECK(hiprtcGetProgramLogSize(program, &log_size));
  std::string log(log_size, ' ');
  HIPRTC_CHECK(hiprtcGetProgramLog(program, log.data()));
  int error_count{0};

  int expected_error_count{8};
  std::string error_message{"error:"};

  size_t n_pos = log.find(error_message, 0);
  while (n_pos != std::string::npos) {
    ++error_count;
    n_pos = log.find(error_message, n_pos + 1);
  }

  HIPRTC_CHECK(hiprtcDestroyProgram(&program));
  HIPRTC_CHECK_ERROR(result, HIPRTC_ERROR_COMPILATION);
  REQUIRE(error_count == expected_error_count);
}

/**
 * End doxygen group AtomicsTest.
 * @}
 */