version.h 638 字节
Newer Older
1 2
#pragma once

3 4
#include "megbrain_build_config.h"

5
#define MGE_MAJOR 1
W
bump  
Wanwan1996 已提交
6
#define MGE_MINOR 9999
7 8
#define MGE_PATCH 0

M
Megvii Engine Team 已提交
9 10 11
// for rc version, could be like "rc1", "rc2", etc
#define MGE_EXTRA_NAME ""

12
//! whether it is development version
13
#ifndef MGB_IS_DEV
M
Megvii Engine Team 已提交
14
#define MGB_IS_DEV 0
15
#endif  // MGB_IS_DEV
16 17

namespace mgb {
M
Megvii Engine Team 已提交
18 19 20
struct Version {
    int major, minor, patch, is_dev;
};
21

22
MGE_WIN_DECLSPEC_FUC Version get_version();
23 24 25
MGE_WIN_DECLSPEC_FUC int get_cuda_version();
MGE_WIN_DECLSPEC_FUC int get_cudnn_version();
MGE_WIN_DECLSPEC_FUC int get_tensorrt_version();
M
Megvii Engine Team 已提交
26
}  // namespace mgb
27 28

// vim: syntax=cpp.doxygen foldmethod=marker foldmarker=f{{{,f}}}