set profiling queue property for dpc++ tests

ginkgo-project · May 23, 2023 · b77bad0 · b77bad0
1 parent 0c7fa28
commit b77bad0
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 2 deletions.
diff --git a/test/base/timer.cpp b/test/base/timer.cpp
@@ -44,7 +44,23 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 #include "test/utils/executor.hpp"
 
 
-class Timer : public CommonTestFixture {};
+class Timer : public CommonTestFixture {
+#ifdef GKO_COMPILING_DPCPP
+    Timer()
+    {
+        // require profiling capability
+        const auto property = dpcpp_queue_property::in_order |
+                              dpcpp_queue_property::enable_profiling;
+        if (gko::DpcppExecutor::get_num_devices("gpu") > 0) {
+            exec = gko::DpcppExecutor::create(0, ref, "gpu", property);
+        } else if (gko::DpcppExecutor::get_num_devices("cpu") > 0) {
+            exec = gko::DpcppExecutor::create(0, ref, "cpu", property);
+        } else {
+            throw std::runtime_error{"No suitable DPC++ devices"};
+        }
+    }
+#endif
+};
 
 
 TEST_F(Timer, WorksAsync)

diff --git a/test/log/profiler_hook.cpp b/test/log/profiler_hook.cpp
@@ -42,7 +42,23 @@ OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 #include "test/utils/executor.hpp"
 
 
-class ProfilerHook : public CommonTestFixture {};
+class ProfilerHook : public CommonTestFixture {
+#ifdef GKO_COMPILING_DPCPP
+    ProfilerHook()
+    {
+        // require profiling capability
+        const auto property = dpcpp_queue_property::in_order |
+                              dpcpp_queue_property::enable_profiling;
+        if (gko::DpcppExecutor::get_num_devices("gpu") > 0) {
+            exec = gko::DpcppExecutor::create(0, ref, "gpu", property);
+        } else if (gko::DpcppExecutor::get_num_devices("cpu") > 0) {
+            exec = gko::DpcppExecutor::create(0, ref, "cpu", property);
+        } else {
+            throw std::runtime_error{"No suitable DPC++ devices"};
+        }
+    }
+#endif
+};
 
 
 void call_ranges_unique(std::shared_ptr<gko::log::ProfilerHook> logger)