xref: /aosp_15_r20/external/pytorch/aten/src/ATen/templates/UfuncCUDA.cu (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #define TORCH_ASSERT_NO_OPERATORS
2 
3 #include <ATen/native/ufunc/${name}.h>
4 #include <ATen/Dispatch.h>
5 #include <ATen/native/DispatchStub.h>
6 #include <c10/core/Scalar.h>
7 ${cuda_headers}
8 
9 namespace at {
10 
11 // NB: this is explicitly copied here (via codegen) rather than
12 // included via NativeFunctions.h to avoid recompiling this file when
13 // NativeFunctions.h changes
14 namespace meta {
15 ${meta_declaration}
16 }
17 
18 namespace native {
19 ${native_declaration}
20 ${native_definitions}
21 }} // namespace at::native
22