Add rocBLAS backend (#144)

* [BLAS] Add rocBLAS backend support through hipSYCL



Co-authored-by: sbalint98 <[email protected]>
Co-authored-by: Mesut Meterelliyoz <[email protected]>
Co-authored-by: Maria Kraynyuk <[email protected]>

Author: 3 people

CMakeLists.txt

@@ -1,5 +1,6 @@
 #===============================================================================
 # Copyright 2020-2021 Intel Corporation
+# Copyright (C) 2022 Heidelberg University, Engineering Mathematics and Computing Lab (EMCL) and Computing Centre (URZ)
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -49,6 +50,7 @@ option(ENABLE_CUBLAS_BACKEND "" OFF)
 
 option(ENABLE_CUSOLVER_BACKEND "" OFF)
 
+option(ENABLE_ROCBLAS_BACKEND "" OFF)
 option(ENABLE_CURAND_BACKEND "" OFF)
 option(ENABLE_NETLIB_BACKEND "" OFF)
 set(ONEMKL_SYCL_IMPLEMENTATION "dpc++" CACHE STRING "Name of the SYCL compiler")

README.md

@@ -35,6 +35,9 @@
 #ifdef ENABLE_CUBLAS_BACKEND
 #include "oneapi/mkl/blas/detail/cublas/blas_ct.hpp"
 #endif
+#ifdef ENABLE_ROCBLAS_BACKEND
+#include "oneapi/mkl/blas/detail/rocblas/blas_ct.hpp"
+#endif
 #ifdef ENABLE_MKLCPU_BACKEND
 #include "oneapi/mkl/blas/detail/mklcpu/blas_ct.hpp"
 #endif

include/oneapi/mkl/blas.hpp

@@ -41,6 +41,9 @@ namespace column_major {
 #define BACKEND cublas
 #include "blas_ct_backends.hxx"
 #undef BACKEND
+#define BACKEND rocblas
+#include "blas_ct_backends.hxx"
+#undef BACKEND
 #define BACKEND netlib
 #include "blas_ct_backends.hxx"
 #undef BACKEND
@@ -57,6 +60,9 @@ namespace row_major {
 #define BACKEND cublas
 #include "blas_ct_backends.hxx"
 #undef BACKEND
+#define BACKEND rocblas
+#include "blas_ct_backends.hxx"
+#undef BACKEND
 #define BACKEND netlib
 #include "blas_ct_backends.hxx"
 #undef BACKEND

include/oneapi/mkl/blas/detail/blas_ct_backends.hpp

@@ -0,0 +1,56 @@
+/***************************************************************************
+*  Copyright (C) Codeplay Software Limited
+*  Copyright (C) 2022 Heidelberg University, Engineering Mathematics and Computing Lab (EMCL) and Computing Centre (URZ)
+*
+*  Licensed under the Apache License, Version 2.0 (the "License");
+*  you may not use this file except in compliance with the License.
+*  You may obtain a copy of the License at
+*
+*      http://www.apache.org/licenses/LICENSE-2.0
+*
+*  For your convenience, a copy of the License has been included in this
+*  repository.
+*
+*  Unless required by applicable law or agreed to in writing, software
+*  distributed under the License is distributed on an "AS IS" BASIS,
+*  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+*  See the License for the specific language governing permissions and
+*  limitations under the License.
+*
+**************************************************************************/
+
+#ifndef _DETAIL_ROCBLAS_BLAS_CT_HPP_
+#define _DETAIL_ROCBLAS_BLAS_CT_HPP_
+
+#include <CL/sycl.hpp>
+#include <complex>
+#include <cstdint>
+
+#include "oneapi/mkl/types.hpp"
+#include "oneapi/mkl/detail/backend_selector.hpp"
+#include "oneapi/mkl/detail/config.hpp"
+#include "oneapi/mkl/blas/detail/rocblas/onemkl_blas_rocblas.hpp"
+#include "oneapi/mkl/blas/detail/blas_ct_backends.hpp"
+
+namespace oneapi {
+namespace mkl {
+namespace blas {
+namespace column_major {
+
+#define MAJOR column_major
+#include "blas_ct.hxx"
+#undef MAJOR
+
+} //namespace column_major
+namespace row_major {
+
+#define MAJOR row_major
+#include "blas_ct.hxx"
+#undef MAJOR
+
+} //namespace row_major
+} //namespace blas
+} //namespace mkl
+} //namespace oneapi
+
+#endif //_DETAIL_ROCBLAS_BLAS_CT_HPP_

include/oneapi/mkl/blas/detail/rocblas/blas_ct.hpp

@@ -0,0 +1,54 @@
+/***************************************************************************
+*  Copyright (C) Codeplay Software Limited
+*  Copyright (C) 2022 Heidelberg University, Engineering Mathematics and Computing Lab (EMCL) and Computing Centre (URZ)
+*
+*  Licensed under the Apache License, Version 2.0 (the "License");
+*  you may not use this file except in compliance with the License.
+*  You may obtain a copy of the License at
+*
+*      http://www.apache.org/licenses/LICENSE-2.0
+*
+*  For your convenience, a copy of the License has been included in this
+*  repository.
+*
+*  Unless required by applicable law or agreed to in writing, software
+*  distributed under the License is distributed on an "AS IS" BASIS,
+*  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+*  See the License for the specific language governing permissions and
+*  limitations under the License.
+*
+**************************************************************************/
+#ifndef _ONEMKL_BLAS_ROCBLAS_HPP_
+#define _ONEMKL_BLAS_ROCBLAS_HPP_
+#include <CL/sycl.hpp>
+#include <complex>
+#include <cstdint>
+#include <string>
+#include "oneapi/mkl/types.hpp"
+#include "oneapi/mkl/detail/config.hpp"
+
+namespace oneapi {
+namespace mkl {
+using oneapi::mkl::diag;
+using oneapi::mkl::offset;
+using oneapi::mkl::side;
+using oneapi::mkl::transpose;
+using oneapi::mkl::uplo;
+namespace blas {
+namespace rocblas {
+namespace column_major {
+
+#include "onemkl_blas_rocblas.hxx"
+
+} //namespace column_major
+namespace row_major {
+
+#include "onemkl_blas_rocblas.hxx"
+
+} //namespace row_major
+} //namespace rocblas
+} //namespace blas
+} //namespace mkl
+} //namespace oneapi
+
+#endif //_ONEMKL_BLAS_ROCBLAS_HPP_