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