1 // Copyright © 2022 Apple Inc. 2 3 #pragma once 4 #include <ATen/core/Tensor.h> 5 6 namespace at::native::mps { 7 8 at::Tensor& mps_copy_( 9 at::Tensor& dst, 10 const at::Tensor& src, 11 bool non_blocking); 12 void copy_blit_mps(void* dst, const void* src, size_t size); 13 14 } // namespace at::native::mps 15