mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-08 07:39:33 +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
26 lines
574 B
C++
26 lines
574 B
C++
// clang-format off
|
|
#pragma once
|
|
#include <ATen/Context.h>
|
|
#include <ATen/DeviceGuard.h>
|
|
#include <ATen/TensorUtils.h>
|
|
#include <ATen/TracerMode.h>
|
|
#include <ATen/core/Generator.h>
|
|
#include <ATen/core/Reduction.h>
|
|
#include <ATen/core/Tensor.h>
|
|
#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>
|
|
// ${generated_comment}
|
|
|
|
${static_dispatch_extra_headers}
|
|
|
|
namespace torch {
|
|
namespace executor {
|
|
|
|
${Functions_declarations}
|
|
|
|
} // namespace executor
|
|
} // namespace torch
|