mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-15 21:00:47 +00:00
a9b0a921d5
Summary: As GoogleTest `TEST` macro is non-compliant with it as well as `DEFINE_DISPATCH` All changes but the ones to `.clang-tidy` are generated using following script: ``` for i in `find . -type f -iname "*.c*" -or -iname "*.h"|xargs grep cppcoreguidelines-avoid-non-const-global-variables|cut -f1 -d:|sort|uniq`; do sed -i "/\/\/ NOLINTNEXTLINE(cppcoreguidelines-avoid-non-const-global-variables)/d" $i; done ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/62008 Reviewed By: driazati, r-barnes Differential Revision: D29838584 Pulled By: malfet fbshipit-source-id: 1b2f8602c945bd4ce50a9bfdd204755556e31d13
38 lines
780 B
C++
38 lines
780 B
C++
#if defined(__SSE3__)
|
|
|
|
#include <immintrin.h>
|
|
|
|
#include "caffe2/core/common.h"
|
|
#include "caffe2/core/init.h"
|
|
|
|
C10_DEFINE_int(
|
|
caffe2_ftz,
|
|
false,
|
|
"If true, turn on flushing denormals to zero (FTZ)");
|
|
C10_DEFINE_int(
|
|
caffe2_daz,
|
|
false,
|
|
"If true, turn on replacing denormals loaded from memory with zero (DAZ)");
|
|
|
|
namespace caffe2 {
|
|
|
|
bool Caffe2SetDenormals(int*, char***) {
|
|
if (FLAGS_caffe2_ftz) {
|
|
VLOG(1) << "Setting FTZ";
|
|
_MM_SET_FLUSH_ZERO_MODE(_MM_FLUSH_ZERO_ON);
|
|
}
|
|
if (FLAGS_caffe2_daz) {
|
|
VLOG(1) << "Setting DAZ";
|
|
_MM_SET_DENORMALS_ZERO_MODE(_MM_DENORMALS_ZERO_ON);
|
|
}
|
|
return true;
|
|
}
|
|
|
|
REGISTER_CAFFE2_INIT_FUNCTION(
|
|
Caffe2SetDenormals,
|
|
&Caffe2SetDenormals,
|
|
"Set denormal settings.");
|
|
|
|
} // namespace caffe2
|
|
|
|
#endif
|