chromium/third_party/xnnpack/src/src/xnnpack/prefetch.h

// Copyright 2022 Google LLC
//
// This source code is licensed under the BSD-style license found in the
// LICENSE file in the root directory of this source tree.

#pragma once

#ifdef _MSC_VER
  #include <intrin.h>  
#endif

#ifdef __hexagon__
  #include <hexagon_protos.h>

  // Use fetchl2 at least several hundred cycles prior to using the data.

  // size_fetch: the number of bytes to fetch.
  XNN_INLINE static void xnn_prefetch_to_l2_linear(void *address, uint32_t size_fetch)
  {
    uint32_t info = HEXAGON_V64_CREATE_H(0, 0, size_fetch, 1);
    Q6_l2fetch_AR(address, info);
  }

  // height: the number of width-sized blocks to fetch.
  // width: width of a fetch block in bytes.
  // stride: an unsigned byte offset which is used to increment the pointer
  //         after each width-sized block is fetched.
  // direction: If clear, row major. If set, cloumn major.
  XNN_INLINE static void xnn_prefetch_to_l2_box(void *address,
                                        uint64_t stride,
                                        uint64_t width,
                                        uint64_t height,
                                        uint64_t direction)
  {
    uint64_t info = HEXAGON_V64_CREATE_H(direction, stride, width, height);
    Q6_l2fetch_AP(address, info);
  }
#endif

#include "xnnpack/common.h"


XNN_INLINE static void xnn_prefetch_to_l1(const void* address) { … }