mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
d06e1df1aa
Rename the context type to match changes in executorch. Differential Revision: [D46977359](https://our.internmc.facebook.com/intern/diff/D46977359/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/104892 Approved by: https://github.com/larryliu0820
22 lines
757 B
C++
22 lines
757 B
C++
#pragma once
|
|
|
|
namespace torch {
|
|
namespace executor {
|
|
|
|
/**
|
|
* Bucket type abstraction that contains many elements of runtime state that
|
|
* a kernel author may want available, but would otherwise be unable to access.
|
|
*
|
|
* Forwarded along to all operators when running in lean mode. NOTE: Will not be
|
|
* forwarded to operators if running in ATen mode as those operators do not
|
|
* expect to receive a KernelRuntimeContext and would not use it.
|
|
*
|
|
* This includes things like setting an error state, a scratch allocator for
|
|
* operators that need more then constant space, and a TensorResizer for dynamic
|
|
* shape tensors allowing programs to be more flexible with Tensor shape.
|
|
*/
|
|
class KernelRuntimeContext {};
|
|
|
|
} // namespace executor
|
|
} // namespace torch
|