cranky-coder08's picture
Add files using upload-large-folder tool
c1af2fa verified
// 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