dmlc--dgl
Hongzhi (Steve), Chen
bcd3768426
* replace * blabla * balbla * blabla Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
33 行
552 B
C++
33 行
552 B
C++
/**
|
|
* Copyright (c) 2019 by Contributors
|
|
* @file runtime/config.h
|
|
* @brief DGL runtime config
|
|
*/
|
|
|
|
#ifndef DGL_RUNTIME_CONFIG_H_
|
|
#define DGL_RUNTIME_CONFIG_H_
|
|
|
|
namespace dgl {
|
|
namespace runtime {
|
|
|
|
class Config {
|
|
public:
|
|
static Config* Global() {
|
|
static Config config;
|
|
return &config;
|
|
}
|
|
|
|
// Enabling or disable use libxsmm for Spmm
|
|
void EnableLibxsmm(bool);
|
|
bool IsLibxsmmAvailable() const;
|
|
|
|
private:
|
|
Config() = default;
|
|
bool libxsmm_ = true;
|
|
};
|
|
|
|
} // namespace runtime
|
|
} // namespace dgl
|
|
|
|
#endif // DGL_RUNTIME_CONFIG_H_
|