mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
fa09099ba3
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/68691 TraceType is a sharded file, so by only including specific operator headers, we ensure that changing one (non-method) operator only needs one shard to be re-compiled. This also changes all the included autograd and jit headers from including `ATen/ATen.h` to just including `ATen/core/Tensor.h`. Test Plan: Imported from OSS Reviewed By: gchanan Differential Revision: D33336948 Pulled By: albanD fbshipit-source-id: 4e40371592b9a5a7e7fcd1d8cecae11ffb873113
14 lines
279 B
C++
14 lines
279 B
C++
#pragma once
|
|
|
|
#include <ATen/core/Tensor.h>
|
|
|
|
namespace torch { namespace autograd {
|
|
|
|
struct TORCH_API SavedVariableHooks {
|
|
virtual void call_pack_hook(const at::Tensor &tensor) = 0;
|
|
virtual at::Tensor call_unpack_hook() = 0;
|
|
virtual ~SavedVariableHooks() = default;
|
|
};
|
|
|
|
}}
|