xref: /libCEED/backends/magma/ceed-magma-gemm-selector.h (revision 6eb06d7cb0f5787c494a4969c0aa6769f3bcfbd0)
1 // Copyright (c) 2017-2025, Lawrence Livermore National Security, LLC and other CEED contributors.
2 // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3 //
4 // SPDX-License-Identifier: BSD-2-Clause
5 //
6 // This file is part of CEED:  http://github.com/ceed
7 #pragma once
8 
9 #include "ceed-magma.h"
10 
11 ////////////////////////////////////////////////////////////////////////////////
12 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);
13 
14 ////////////////////////////////////////////////////////////////////////////////
15 CEED_INTERN CeedInt nontensor_rtc_get_nb(int gpu_arch, char trans_A, int q_comp, int P, int Q, int N);
16