mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
2f154f68ea
## Job Test running on most CI jobs. ## Test binary * `test_main.cpp`: entry for gtest * `test_operator_registration.cpp`: test cases for gtest ## Helper sources * `operator_registry.h/cpp`: simple operator registry for testing purpose. * `Evalue.h`: a boxed data type that wraps ATen types, for testing purpose. * `selected_operators.yaml`: operators Executorch care about so far, we should cover all of them. ## Templates * `NativeFunctions.h`: for generating headers for native functions. (not compiled in the test, since we will be using `libtorch`) * `RegisterCodegenUnboxedKernels.cpp`: for registering boxed operators. * `Functions.h`: for declaring operator C++ APIs. Generated `Functions.h` merely wraps `ATen/Functions.h`. ## Build files * `CMakeLists.txt`: generate code to register ops. * `build.sh`: driver file, to be called by CI job. Pull Request resolved: https://github.com/pytorch/pytorch/pull/89596 Approved by: https://github.com/ezyang
32 lines
1.1 KiB
C++
32 lines
1.1 KiB
C++
#pragma once
|
|
|
|
// ${generated_comment}
|
|
|
|
#ifdef TORCH_ASSERT_NO_OPERATORS
|
|
#error This change adds a dependency on native_functions.yaml, \
|
|
meaning the file will need to be re-compiled every time an operator \
|
|
is changed or added. Consider if your change would be better placed in \
|
|
another file, or if a more specific header might achieve the same goal. \
|
|
See NOTE: [Tensor vs. TensorBase]
|
|
#endif
|
|
|
|
#if defined(AT_PER_OPERATOR_HEADERS) && defined(TORCH_ASSERT_ONLY_METHOD_OPERATORS)
|
|
#error This change adds a dependency on all pytorch operators, meaning the \
|
|
file will need to be re-compiled every time an operator is changed or added. \
|
|
Consider including a specific operator from <ATen/ops/{my_operator}_native.h> \
|
|
and see NOTE [TORCH_ASSERT_ONLY_METHOD_OPERATORS].
|
|
#endif
|
|
|
|
#include <c10/core/Scalar.h>
|
|
#include <c10/core/Storage.h>
|
|
#include <c10/core/TensorOptions.h>
|
|
#include <c10/util/Deprecated.h>
|
|
#include <c10/util/Optional.h>
|
|
#include <c10/core/QScheme.h>
|
|
#include <ATen/core/Reduction.h>
|
|
#include <ATen/core/Tensor.h>
|
|
#include <tuple>
|
|
#include <vector>
|
|
|
|
${nativeFunctions_declarations}
|