summaryrefslogtreecommitdiffstats
path: root/contrib/libs/cxxsupp/builtins/aarch64/sme-libc-memcpy-memmove.c
blob: 9792079964d55b07e8af0896acd5c01969e1da67 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
///
/// \file
/// This file contains basic implementations of Scalable Matrix Extension (SME)
/// compatible memcpy and memmove functions to be used when their assembly-
/// optimized counterparts can't.
///
//===----------------------------------------------------------------------===//

#include <stddef.h>

static void *__arm_sc_memcpy_fwd(void *dest, const void *src,
                                 size_t n) __arm_streaming_compatible {
  unsigned char *destp = (unsigned char *)dest;
  const unsigned char *srcp = (const unsigned char *)src;

  for (size_t i = 0; i < n; ++i)
    destp[i] = srcp[i];
  return dest;
}

static void *__arm_sc_memcpy_rev(void *dest, const void *src,
                                 size_t n) __arm_streaming_compatible {
  unsigned char *destp = (unsigned char *)dest;
  const unsigned char *srcp = (const unsigned char *)src;

  while (n > 0) {
    --n;
    destp[n] = srcp[n];
  }
  return dest;
}

extern void *__arm_sc_memcpy(void *__restrict dest, const void *__restrict src,
                             size_t n) __arm_streaming_compatible {
  return __arm_sc_memcpy_fwd(dest, src, n);
}

extern void *__arm_sc_memmove(void *dest, const void *src,
                              size_t n) __arm_streaming_compatible {
  unsigned char *destp = (unsigned char *)dest;
  const unsigned char *srcp = (const unsigned char *)src;

  if ((srcp > (destp + n)) || (destp > (srcp + n)))
    return __arm_sc_memcpy(dest, src, n);
  if (srcp > destp)
    return __arm_sc_memcpy_fwd(dest, src, n);
  return __arm_sc_memcpy_rev(dest, src, n);
}