dmlc--dgl
peizhou001
62af41c245
* enable turn on/off libxsmm at runtime by adding a global config and related API Co-authored-by: Ubuntu <ubuntu@ip-172-31-19-194.ap-northeast-1.compute.internal>
34 行
554 B
C++
34 行
554 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_
|