| #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 RuntimeContext 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 RuntimeContext {}; |
| |
| } // namespace executor |
| } // namespace torch |
