#pragma once
|
|
// Detect address sanitizer as some stuff doesn't work with it
|
|
#undef CAFFE2_ASAN_ENABLED
|
|
// for clang
|
#if defined(__has_feature)
|
#if ((__has_feature(address_sanitizer)))
|
#define CAFFE2_ASAN_ENABLED 1
|
#endif
|
#endif
|
|
// for gcc
|
#if defined(__SANITIZE_ADDRESS__)
|
#if __SANITIZE_ADDRESS__
|
#if !defined(CAFFE2_ASAN_ENABLED)
|
#define CAFFE2_ASAN_ENABLED 1
|
#endif
|
#endif
|
#endif
|
|
#if !defined(CAFFE2_ASAN_ENABLED)
|
#define CAFFE2_ASAN_ENABLED 0
|
#endif
|
|
// Define sanitization macro
|
#if !CAFFE2_ASAN_ENABLED
|
#define CAFFE2_NO_SANITIZE(...)
|
#else
|
#define CAFFE2_NO_SANITIZE(...) __attribute__((no_sanitize(__VA_ARGS__)))
|
#endif
|
