| 1234567891011121314151617181920212223242526272829 |
- // We register ops with a higher priority dispatch key (BackendSelect) than the usual backend-specific keys (e.g. CPU)
- // which makes calls to the factory functions dispatch to here.
- // We then 'manually' compute a lower-priority to re-dispatch to (e.g. CPU) to get to the eventually correct backend.
- // ${generated_comment}
- #define TORCH_ASSERT_ONLY_METHOD_OPERATORS
- #include <ATen/core/Tensor.h>
- #include <ATen/core/dispatch/DispatchKeyExtractor.h>
- #include <torch/library.h>
- #ifndef AT_PER_OPERATOR_HEADERS
- #include <ATen/Operators.h>
- #else
- ${ops_headers}
- #endif
- namespace at {
- namespace {
- ${backend_select_method_definitions}
- TORCH_LIBRARY_IMPL(aten, BackendSelect, m) {
- ${backend_select_function_registrations};
- }
- } // namespace
- } // at
|
