xref: /libCEED/rust/libceed-sys/c-src/backends/hip/ceed-hip-compile.h (revision 3d8e882215d238700cdceb37404f76ca7fa24eaa)
1*3d8e8822SJeremy L Thompson // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors.
2*3d8e8822SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
330f4f45fSnbeams //
4*3d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause
530f4f45fSnbeams //
6*3d8e8822SJeremy L Thompson // This file is part of CEED:  http://github.com/ceed
73d576824SJeremy L Thompson 
830f4f45fSnbeams #ifndef _ceed_hip_compile_h
930f4f45fSnbeams #define _ceed_hip_compile_h
1030f4f45fSnbeams 
11ec3da8bcSJed Brown #include <ceed/ceed.h>
1230f4f45fSnbeams #include <hip/hip_runtime.h>
1330f4f45fSnbeams 
140d0321e0SJeremy L Thompson static inline CeedInt CeedDivUpInt(CeedInt numerator, CeedInt denominator) {
150d0321e0SJeremy L Thompson   return (numerator + denominator - 1) / denominator;
160d0321e0SJeremy L Thompson }
170d0321e0SJeremy L Thompson 
18cf7477b8Snbeams CEED_INTERN int CeedCompileHip(Ceed ceed, const char *source,
190d0321e0SJeremy L Thompson                                hipModule_t *module, const CeedInt numopts, ...);
2030f4f45fSnbeams 
21cf7477b8Snbeams CEED_INTERN int CeedGetKernelHip(Ceed ceed, hipModule_t module,
220d0321e0SJeremy L Thompson                                  const char *name, hipFunction_t *kernel);
2330f4f45fSnbeams 
2430f4f45fSnbeams CEED_INTERN int CeedRunKernelHip(Ceed ceed, hipFunction_t kernel,
250d0321e0SJeremy L Thompson                                  const int grid_size,
260d0321e0SJeremy L Thompson                                  const int block_size, void **args);
2730f4f45fSnbeams 
2830f4f45fSnbeams CEED_INTERN int CeedRunKernelDimHip(Ceed ceed, hipFunction_t kernel,
290d0321e0SJeremy L Thompson                                     const int grid_size,
300d0321e0SJeremy L Thompson                                     const int block_size_x, const int block_size_y,
310d0321e0SJeremy L Thompson                                     const int block_size_z, void **args);
3230f4f45fSnbeams 
3330f4f45fSnbeams CEED_INTERN int CeedRunKernelDimSharedHip(Ceed ceed, hipFunction_t kernel,
340d0321e0SJeremy L Thompson     const int grid_size, const int block_size_x, const int block_size_y,
350d0321e0SJeremy L Thompson     const int block_size_z, const int shared_mem_size, void **args);
3630f4f45fSnbeams 
373d576824SJeremy L Thompson #endif // _ceed_hip_compile_h
38