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