xref: /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/CopyKernel.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1 #pragma once
2 
3 #include <ATen/native/TensorIterator.h>
4 
5 namespace at {
6 struct TensorIteratorBase;
7 
8 namespace native {
9 inline namespace CPU_CAPABILITY {
10 
11 void direct_copy_kernel(TensorIteratorBase &iter);
12 void copy_kernel(TensorIterator& iter, bool /*non_blocking*/);
13 
14 }}}  // namespace at::native::CPU_CAPABILITY
15