config.h 545 Bytes
Newer Older
1 2
/*!
 *  Copyright (c) 2017 by Contributors
tqchen committed
3
 * \file tvm/runtime/config.h
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22
 * \brief Runtime library related configurations.
 */
#ifndef TVM_RUNTIME_CONFIG_H_
#define TVM_RUNTIME_CONFIG_H_

/*!
 *\brief whether to use CUDA runtime
 */
#ifndef TVM_CUDA_RUNTIME
#define TVM_CUDA_RUNTIME 1
#endif

/*!
 *\brief whether to use opencl runtime
 */
#ifndef TVM_OPENCL_RUNTIME
#define TVM_OPENCL_RUNTIME 0
#endif

23 24 25 26 27 28 29
/*!
 *\brief whether to use metal runtime
 */
#ifndef TVM_METAL_RUNTIME
#define TVM_METAL_RUNTIME 0
#endif

30
#endif  // TVM_RUNTIME_CONFIG_H_