dmlc--dgl
Daniil Sizov
87fb7ed05b
* Enable AVX by default * Fix linting errors * Fix win64 build (libxsmm not linked) Libxsmm on Win64 is not linked, should be disabled by default * Fix clang format issues * Change lower supported cpu version to LIBXSMM_X86_AVX2 Change lower supported cpu version to LIBXSMM_X86_AVX2 to address https://github.com/dmlc/dgl/issues/3459 issue * Fix unit test Remove assumption that libxsmm is enabled in the config by default (only true for intel CPUs with AVX2 instructions) --------- Co-authored-by: Ubuntu <ubuntu@ip-172-31-15-137.us-west-2.compute.internal> Co-authored-by: Quan (Andy) Gan <coin2028@hotmail.com>
33 行
535 B
C++
33 行
535 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();
|
|
bool libxsmm_;
|
|
};
|
|
|
|
} // namespace runtime
|
|
} // namespace dgl
|
|
|
|
#endif // DGL_RUNTIME_CONFIG_H_
|