xref: /libCEED/backends/magma/ceed-magma-gemm-selector.h (revision d4cc18453651bd0f94c1a2e078b2646a92dafdcc)
1*9ba83ac0SJeremy L Thompson // Copyright (c) 2017-2026, Lawrence Livermore National Security, LLC and other CEED contributors.
2940a72f1SSebastian Grimberg // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
3940a72f1SSebastian Grimberg //
4940a72f1SSebastian Grimberg // SPDX-License-Identifier: BSD-2-Clause
5940a72f1SSebastian Grimberg //
6940a72f1SSebastian Grimberg // This file is part of CEED:  http://github.com/ceed
7509d4af6SJeremy L Thompson #pragma once
8940a72f1SSebastian Grimberg 
9940a72f1SSebastian Grimberg #include "ceed-magma.h"
10940a72f1SSebastian Grimberg 
11940a72f1SSebastian Grimberg ////////////////////////////////////////////////////////////////////////////////
12940a72f1SSebastian 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);
13940a72f1SSebastian Grimberg 
14940a72f1SSebastian Grimberg ////////////////////////////////////////////////////////////////////////////////
157c7f2ed8SSebastian Grimberg CEED_INTERN CeedInt nontensor_rtc_get_nb(int gpu_arch, char trans_A, int q_comp, int P, int Q, int N);
16