runtime.h 770 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
/*!
 * \file tl/runtime/runtime.h
 * \brief Runtime functions.
 *
 */

#ifndef TVM_TL_RUNTIME_RUNTIME_H_
#define TVM_TL_RUNTIME_RUNTIME_H_

namespace tvm {
namespace tl {

13
#if (CUDA_MAJOR_VERSION >= 12)
14
15
16
17
constexpr const char *tvm_tensormap_create_tiled =
    "__tvm_tensormap_create_tiled";
constexpr const char *tvm_tensormap_create_im2col =
    "__tvm_tensormap_create_im2col";
18
#endif // (CUDA_MAJOR_VERSION >= 12)
19
20
21
22
23
24

// CUDA stream access policy window helpers
constexpr const char *tvm_cuda_stream_set_access_policy_window =
    "__tvm_cuda_stream_set_access_policy_window";
constexpr const char *tvm_cuda_stream_reset_access_policy_window =
    "__tvm_cuda_stream_reset_access_policy_window";
25
26
} // namespace tl
} // namespace tvm
27

28
#endif //  TVM_TL_RUNTIME_RUNTIME_H_