EXSWHTEC-251 - Implement Performance Tests for Stream APIs (#142)

Change-Id: I790d8cadb3a94a220a52c8bdc0a23b1a63931232

catch/performance/stream/hipMallocFromPoolAsync.cc

@@ -0,0 +1,82 @@
+/*
+Copyright (c) 2022 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.
+*/
+
+#include "mem_pools_performance_common.hh"
+
+/**
+ * @addtogroup stream stream
+ * @{
+ * @ingroup PerformanceTest
+ */
+
+class MallocFromPoolAsyncBenchmark : public Benchmark<MallocFromPoolAsyncBenchmark> {
+ public:
+  void operator()(const size_t array_size) {
+    const StreamGuard stream_guard{Streams::created};
+    const hipStream_t stream = stream_guard.stream();
+
+    hipMemPool_t mem_pool{nullptr};
+    hipMemPoolProps pool_props = CreateMemPoolProps(0, hipMemHandleTypeNone);
+    HIP_CHECK(hipMemPoolCreate(&mem_pool, &pool_props));
+
+    float* array_ptr{nullptr};
+
+    TIMED_SECTION_STREAM(kTimerTypeEvent, stream) {
+      HIP_CHECK(hipMallocFromPoolAsync(&array_ptr, array_size * sizeof(float), mem_pool, stream));
+    }
+
+    REQUIRE(array_ptr != nullptr);
+
+    HIP_CHECK(hipFreeAsync(array_ptr, stream));
+    HIP_CHECK(hipStreamSynchronize(stream));
+    HIP_CHECK(hipMemPoolDestroy(mem_pool));
+  }
+};
+
+static void RunBenchmark(const size_t array_size) {
+  MallocFromPoolAsyncBenchmark benchmark;
+  benchmark.AddSectionName(std::to_string(array_size));
+  benchmark.Run(array_size);
+}
+
+/**
+ * Test Description
+ * ------------------------
+ *  - Executes `hipMallocFromPoolAsync`:
+ *    -# Allocation size:
+ *      - 4 KB
+ *      - 4 MB
+ *      - 16 MB
+ * Test source
+ * ------------------------
+ *  - performance/stream/hipMallocFromPoolAsync.cc
+ * Test requirements
+ * ------------------------
+ *  - Device supports memory pools
+ *  - HIP_VERSION >= 5.2
+ */
+TEST_CASE("Performance_hipMallocFromPoolAsync") {
+  if (!AreMemPoolsSupported(0)) {
+    HipTest::HIP_SKIP_TEST("GPU 0 doesn't support hipDeviceAttributeMemoryPoolsSupported "
+                           "attribute. Hence skipping the testing with Pass result.\n");
+    return;
+  }
+  size_t array_size = GENERATE(4_KB, 4_MB, 16_MB);
+  RunBenchmark(array_size);
+}