1*940a72f1SSebastian Grimberg // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors. 2*940a72f1SSebastian Grimberg // All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 3*940a72f1SSebastian Grimberg // 4*940a72f1SSebastian Grimberg // SPDX-License-Identifier: BSD-2-Clause 5*940a72f1SSebastian Grimberg // 6*940a72f1SSebastian Grimberg // This file is part of CEED: http://github.com/ceed 7*940a72f1SSebastian Grimberg 8*940a72f1SSebastian Grimberg #ifndef CEED_MAGMA_GEMM_SELECTOR_H 9*940a72f1SSebastian Grimberg #define CEED_MAGMA_GEMM_SELECTOR_H 10*940a72f1SSebastian Grimberg 11*940a72f1SSebastian Grimberg #include "ceed-magma.h" 12*940a72f1SSebastian Grimberg 13*940a72f1SSebastian Grimberg //////////////////////////////////////////////////////////////////////////////// 14*940a72f1SSebastian Grimberg CEED_INTERN void gemm_selector(int gpu_arch, char precision, char trans_A, int m, int n, int k, int *n_batch, int *use_magma); 15*940a72f1SSebastian Grimberg 16*940a72f1SSebastian Grimberg //////////////////////////////////////////////////////////////////////////////// 17*940a72f1SSebastian Grimberg CEED_INTERN CeedInt nontensor_rtc_get_nb(int gpu_arch, char trans_A, int q_comp, int P, int Q, int n); 18*940a72f1SSebastian Grimberg 19*940a72f1SSebastian Grimberg #endif // CEED_MAGMA_GEMM_SELECTOR_H 20