mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
8f0603b128
Summary: * adds TORCH_API and AT_CUDA_API in places * refactor code generation Python logic to separate caffe2/torch outputs * fix hip and asan * remove profiler_cuda from hip * fix gcc warnings for enums * Fix PythonOp::Kind Pull Request resolved: https://github.com/pytorch/pytorch/pull/19554 Differential Revision: D15082727 Pulled By: kostmo fbshipit-source-id: 83a8a99717f025ab44b29608848928d76b3147a4
27 lines
434 B
C++
27 lines
434 B
C++
#pragma once
|
|
|
|
#include <torch/csrc/WindowsTorchApiMacro.h>
|
|
|
|
namespace torch { namespace autograd {
|
|
|
|
struct TORCH_API AnomalyMode {
|
|
static bool is_enabled() {
|
|
return _enabled;
|
|
}
|
|
static void set_enabled(bool enabled) {
|
|
_enabled = enabled;
|
|
}
|
|
|
|
private:
|
|
static bool _enabled;
|
|
};
|
|
|
|
|
|
struct TORCH_API AnomalyMetadata {
|
|
virtual ~AnomalyMetadata();
|
|
virtual void store_stack() = 0;
|
|
virtual void print_stack() = 0;
|
|
};
|
|
|
|
}}
|