22 lines
494 B
Plaintext
22 lines
494 B
Plaintext
|
#define TORCH_ASSERT_NO_OPERATORS
|
||
|
|
|
||
|
|
#include <ATen/native/ufunc/${name}.h>
|
||
|
|
#include <ATen/Dispatch.h>
|
||
|
|
#include <ATen/native/DispatchStub.h>
|
||
|
|
#include <c10/core/Scalar.h>
|
||
|
|
${cuda_headers}
|
||
|
|
|
||
|
|
namespace at {
|
||
|
|
|
||
|
|
// NB: this is explicitly copied here (via codegen) rather than
|
||
|
|
// included via NativeFunctions.h to avoid recompiling this file when
|
||
|
|
// NativeFunctions.h changes
|
||
|
|
namespace meta {
|
||
|
|
${meta_declaration}
|
||
|
|
}
|
||
|
|
|
||
|
|
namespace native {
|
||
|
|
${native_declaration}
|
||
|
|
${native_definitions}
|
||
|
|
}} // namespace at::native
|