#if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
//  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);

// Vectorized same-dtype scatter of a CONTIGUOUS `input` into regularly-strided
// blocks of `output`; see StridedBlockParams (kernels/Copy.h) for the byte
// geometry. Backs the contiguous cat fast path.
void inner_contiguous_scatter_mps(
    const at::Tensor& input,
    const at::Tensor& output,
    uint64_t slice_bytes,
    uint64_t out_stride_bytes,
    uint64_t off_bytes);

} // namespace at::native::mps

#else
#error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
#endif  // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)