add get_compute_capability in cudaExecutor

Co-authored-by: Pratik Nayak <pratikvn@protonmail.com>

Author: yhmtsai

common/cuda_hip/matrix/coo_kernels.cpp

@@ -276,8 +276,8 @@ void spmv2(std::shared_ptr<const DefaultExecutor> exec,
         GKO_NOT_SUPPORTED(c);
     } else
 #else
-    auto compute_capability = as<CudaExecutor>(exec)->get_major_version() * 10 +
-                              as<CudaExecutor>(exec)->get_minor_version();
+    const auto compute_capability =
+        as<CudaExecutor>(exec)->get_compute_capability();
     if (compute_capability < 70 &&
         std::is_same_v<remove_complex<ValueType>, half>) {
         GKO_NOT_SUPPORTED(c);
@@ -338,8 +338,8 @@ void advanced_spmv2(std::shared_ptr<const DefaultExecutor> exec,
         GKO_NOT_SUPPORTED(c);
     } else
 #else
-    auto compute_capability = as<CudaExecutor>(exec)->get_major_version() * 10 +
-                              as<CudaExecutor>(exec)->get_minor_version();
+    const auto compute_capability =
+        as<CudaExecutor>(exec)->get_compute_capability();
     if (compute_capability < 70 &&
         std::is_same_v<remove_complex<ValueType>, half>) {
         GKO_NOT_SUPPORTED(c);

common/cuda_hip/matrix/csr_kernels.template.cpp

@@ -2097,8 +2097,7 @@ bool load_balance_spmv(std::shared_ptr<const DefaultExecutor> exec,
         return false;
     } else
 #else
-    auto compute_capability = as<CudaExecutor>(exec)->get_major_version() * 10 +
-                              as<CudaExecutor>(exec)->get_minor_version();
+    auto compute_capability = as<CudaExecutor>(exec)->get_compute_capability();
     if (compute_capability < 70 &&
         std::is_same_v<remove_complex<OutputValueType>, half>) {
         return false;

common/cuda_hip/matrix/ell_kernels.cpp

@@ -272,8 +272,8 @@ void abstract_spmv(syn::value_list<int, info>,
         sizeof(remove_complex<arithmetic_type>) == sizeof(bfloat16);
     constexpr bool atomic_bfloat16_out =
         atomic && sizeof(remove_complex<OutputValueType>) == sizeof(bfloat16);
-    auto compute_capability = as<CudaExecutor>(exec)->get_major_version() * 10 +
-                              as<CudaExecutor>(exec)->get_minor_version();
+    const auto compute_capability =
+        as<CudaExecutor>(exec)->get_compute_capability();
     if (num_thread_per_worker != 1 && (shared_half || atomic_half_out) &&
         compute_capability < 70) {
         GKO_KERNEL_NOT_FOUND;

common/cuda_hip/solver/idr_kernels.cpp

@@ -460,9 +460,8 @@ void update_g_and_u(std::shared_ptr<const DefaultExecutor> exec,
                 GKO_NOT_SUPPORTED(alpha);
             } else
 #else
-            auto compute_capability =
-                as<CudaExecutor>(exec)->get_major_version() * 10 +
-                as<CudaExecutor>(exec)->get_minor_version();
+            const auto compute_capability =
+                as<CudaExecutor>(exec)->get_compute_capability();
             if (compute_capability < 70 &&
                 std::is_same_v<remove_complex<ValueType>, half>) {
                 GKO_NOT_SUPPORTED(alpha);
@@ -531,9 +530,8 @@ void update_m(std::shared_ptr<const DefaultExecutor> exec, const size_type nrhs,
                 GKO_NOT_SUPPORTED(m_i);
             } else
 #else
-            auto compute_capability =
-                as<CudaExecutor>(exec)->get_major_version() * 10 +
-                as<CudaExecutor>(exec)->get_minor_version();
+            const auto compute_capability =
+                as<CudaExecutor>(exec)->get_compute_capability();
             if (compute_capability < 70 &&
                 std::is_same_v<remove_complex<ValueType>, half>) {
                 GKO_NOT_SUPPORTED(m_i);

include/ginkgo/core/base/executor.hpp

@@ -1,4 +1,4 @@
-// SPDX-FileCopyrightText: 2017 - 2024 The Ginkgo authors
+// SPDX-FileCopyrightText: 2017 - 2025 The Ginkgo authors
 //
 // SPDX-License-Identifier: BSD-3-Clause
 
@@ -1656,6 +1656,14 @@ class CudaExecutor : public detail::ExecutorBase<CudaExecutor>,
         return this->get_exec_info().minor;
     }
 
+    /**
+     * Get the compute capability
+     */
+    int get_compute_capability() const noexcept
+    {
+        return this->get_major_version() * 10 + this->get_minor_version();
+    }
+
     /**
      * Get the cublas handle for this executor
      *