mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
26 lines
459 B
C
26 lines
459 B
C
#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
|