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