Skip to content

Commit f1929dc

Browse files
author
Jenkins
committed
Compute Library v24.08
1 parent c5dd775 commit f1929dc

File tree

858 files changed

+148201
-120545
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

858 files changed

+148201
-120545
lines changed

Android.bp

Lines changed: 26 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -330,9 +330,11 @@ cc_library_static {
330330
"src/core/NEON/kernels/arm_gemm/gemm_fp32.cpp",
331331
"src/core/NEON/kernels/arm_gemm/gemm_int16.cpp",
332332
"src/core/NEON/kernels/arm_gemm/gemm_int8.cpp",
333+
"src/core/NEON/kernels/arm_gemm/gemm_q8_mixed.cpp",
333334
"src/core/NEON/kernels/arm_gemm/gemm_qint8.cpp",
334335
"src/core/NEON/kernels/arm_gemm/gemm_quint8.cpp",
335336
"src/core/NEON/kernels/arm_gemm/gemm_s8fp32.cpp",
337+
"src/core/NEON/kernels/arm_gemm/gemm_u8s8fp32.cpp",
336338
"src/core/NEON/kernels/arm_gemm/gemm_uint16.cpp",
337339
"src/core/NEON/kernels/arm_gemm/gemm_uint8.cpp",
338340
"src/core/NEON/kernels/arm_gemm/interleave-8way.cpp",
@@ -452,6 +454,7 @@ cc_library_static {
452454
"src/cpu/kernels/CpuQuantizeKernel.cpp",
453455
"src/cpu/kernels/CpuReshapeKernel.cpp",
454456
"src/cpu/kernels/CpuScaleKernel.cpp",
457+
"src/cpu/kernels/CpuScatterKernel.cpp",
455458
"src/cpu/kernels/CpuSoftmaxKernel.cpp",
456459
"src/cpu/kernels/CpuSubKernel.cpp",
457460
"src/cpu/kernels/CpuTransposeKernel.cpp",
@@ -501,6 +504,10 @@ cc_library_static {
501504
"src/cpu/kernels/directconv2d/nhwc/neon/fp32.cpp",
502505
"src/cpu/kernels/directconv2d/nhwc/neon/impl.cpp",
503506
"src/cpu/kernels/directconv2d/nhwc/neon/qasymm8.cpp",
507+
"src/cpu/kernels/directconv2d_output_stage/generic/neon/fp16.cpp",
508+
"src/cpu/kernels/directconv2d_output_stage/generic/neon/fp32.cpp",
509+
"src/cpu/kernels/directconv2d_output_stage/generic/neon/qasymm8.cpp",
510+
"src/cpu/kernels/directconv2d_output_stage/generic/neon/qasymm8_signed.cpp",
504511
"src/cpu/kernels/elementwise_binary/generic/neon/fp16.cpp",
505512
"src/cpu/kernels/elementwise_binary/generic/neon/fp32.cpp",
506513
"src/cpu/kernels/elementwise_binary/generic/neon/integer.cpp",
@@ -624,6 +631,7 @@ cc_library_static {
624631
"src/cpu/operators/CpuQuantize.cpp",
625632
"src/cpu/operators/CpuReshape.cpp",
626633
"src/cpu/operators/CpuScale.cpp",
634+
"src/cpu/operators/CpuScatter.cpp",
627635
"src/cpu/operators/CpuSoftmax.cpp",
628636
"src/cpu/operators/CpuSub.cpp",
629637
"src/cpu/operators/CpuTranspose.cpp",
@@ -989,6 +997,7 @@ cc_library_static {
989997
"src/runtime/NEON/functions/NEReshapeLayer.cpp",
990998
"src/runtime/NEON/functions/NEReverse.cpp",
991999
"src/runtime/NEON/functions/NEScale.cpp",
1000+
"src/runtime/NEON/functions/NEScatter.cpp",
9921001
"src/runtime/NEON/functions/NESelect.cpp",
9931002
"src/runtime/NEON/functions/NESlice.cpp",
9941003
"src/runtime/NEON/functions/NESoftmaxLayer.cpp",
@@ -1014,10 +1023,16 @@ cc_library_static {
10141023
"src/runtime/Tensor.cpp",
10151024
"src/runtime/TensorAllocator.cpp",
10161025
"src/runtime/Utils.cpp",
1026+
"src/runtime/experimental/low_level/CpuGemmAssemblyDispatch.cpp",
10171027
"src/runtime/experimental/operators/CpuActivation.cpp",
1028+
"src/runtime/experimental/operators/CpuAdd.cpp",
1029+
"src/runtime/experimental/operators/CpuDepthwiseConv2d.cpp",
1030+
"src/runtime/experimental/operators/CpuElementwise.cpp",
10181031
"src/runtime/experimental/operators/CpuGemm.cpp",
10191032
"src/runtime/experimental/operators/CpuGemmConv2d.cpp",
10201033
"src/runtime/experimental/operators/CpuGemmDirectConv2d.cpp",
1034+
"src/runtime/experimental/operators/CpuMul.cpp",
1035+
"src/runtime/experimental/operators/CpuSub.cpp",
10211036
"src/runtime/experimental/operators/CpuTranspose.cpp",
10221037
"src/runtime/experimental/operators/CpuWinogradConv2d.cpp",
10231038
"src/runtime/heuristics/direct_conv/ClDirectConvDefaultConfigBifrost.cpp",
@@ -1278,6 +1293,10 @@ cc_library_static {
12781293
"src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8qa_dot_4x16/a55.cpp",
12791294
"src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8qa_dot_4x16/generic.cpp",
12801295
"src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8qa_mmla_4x16/generic.cpp",
1296+
"src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8s8qa_dot_4x16/generic.cpp",
1297+
"src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8s8qa_mmla_4x16/generic.cpp",
1298+
"src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8s8s32_dot_6x16/generic.cpp",
1299+
"src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8s8s32_mmla_6x16/generic.cpp",
12811300
"src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8u32_dot_6x16/a55.cpp",
12821301
"src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8u32_dot_6x16/generic.cpp",
12831302
"src/core/NEON/kernels/arm_gemm/kernels/a64_hybrid_u8u32_mmla_6x16/generic.cpp",
@@ -1287,6 +1306,7 @@ cc_library_static {
12871306
"src/core/NEON/kernels/arm_gemm/kernels/a64_interleaved_bf16fp32_mmla_8x12/generic.cpp",
12881307
"src/core/NEON/kernels/arm_gemm/kernels/a64_interleaved_s8s32_mmla_8x12/a510.cpp",
12891308
"src/core/NEON/kernels/arm_gemm/kernels/a64_interleaved_s8s32_mmla_8x12/generic.cpp",
1309+
"src/core/NEON/kernels/arm_gemm/kernels/a64_interleaved_u8s8s32_mmla_8x12/generic.cpp",
12901310
"src/core/NEON/kernels/arm_gemm/kernels/a64_interleaved_u8u32_mmla_8x12/a510.cpp",
12911311
"src/core/NEON/kernels/arm_gemm/kernels/a64_interleaved_u8u32_mmla_8x12/generic.cpp",
12921312
"src/core/NEON/kernels/arm_gemm/kernels/a64_sgemm_8x12/a53.cpp",
@@ -1307,6 +1327,7 @@ cc_library_static {
13071327
"src/core/NEON/kernels/arm_gemm/kernels/a64_smallK_hybrid_u8u32_dot_8x4/a55.cpp",
13081328
"src/core/NEON/kernels/arm_gemm/kernels/a64_smallK_hybrid_u8u32_dot_8x4/generic.cpp",
13091329
"src/core/NEON/kernels/arm_gemm/kernels/sme2_gemv_bf16fp32_dot_16VL/generic.cpp",
1330+
"src/core/NEON/kernels/arm_gemm/kernels/sme2_gemv_fp16_mla_16VL/generic.cpp",
13101331
"src/core/NEON/kernels/arm_gemm/kernels/sme2_gemv_fp16fp32fp16_dot_16VL/generic.cpp",
13111332
"src/core/NEON/kernels/arm_gemm/kernels/sme2_gemv_fp32_mla_16VL/generic.cpp",
13121333
"src/core/NEON/kernels/arm_gemm/kernels/sme2_gemv_fp32bf16fp32_dot_16VL/generic.cpp",
@@ -1339,6 +1360,7 @@ cc_library_static {
13391360
"src/core/NEON/kernels/arm_gemm/kernels/sve_ffhybrid_fp32_mla_6x4VL/a64fx.cpp",
13401361
"src/core/NEON/kernels/arm_gemm/kernels/sve_ffhybrid_fp32_mla_6x4VL/generic.cpp",
13411362
"src/core/NEON/kernels/arm_gemm/kernels/sve_ffhybrid_fp32bf16fp32_mmla_4x6VL/generic.cpp",
1363+
"src/core/NEON/kernels/arm_gemm/kernels/sve_ffinterleaved_bf16fp32_dot_8x3VL/generic.cpp",
13421364
"src/core/NEON/kernels/arm_gemm/kernels/sve_ffinterleaved_bf16fp32_mmla_8x3VL/generic.cpp",
13431365
"src/core/NEON/kernels/arm_gemm/kernels/sve_ffinterleaved_fp16_mla_8x3VL/a64fx.cpp",
13441366
"src/core/NEON/kernels/arm_gemm/kernels/sve_ffinterleaved_fp16_mla_8x3VL/generic.cpp",
@@ -1363,6 +1385,9 @@ cc_library_static {
13631385
"src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_s8s32_mmla_6x4VL/generic.cpp",
13641386
"src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_u8qa_dot_4x4VL/generic.cpp",
13651387
"src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_u8qa_mmla_4x4VL/generic.cpp",
1388+
"src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_u8s8qa_dot_4x4VL/generic.cpp",
1389+
"src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_u8s8qa_mmla_4x4VL/generic.cpp",
1390+
"src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_u8s8s32_mmla_6x4VL/generic.cpp",
13661391
"src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_u8u32_dot_6x4VL/a64fx.cpp",
13671392
"src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_u8u32_dot_6x4VL/generic.cpp",
13681393
"src/core/NEON/kernels/arm_gemm/kernels/sve_hybrid_u8u32_mmla_6x4VL/generic.cpp",
@@ -1376,6 +1401,7 @@ cc_library_static {
13761401
"src/core/NEON/kernels/arm_gemm/kernels/sve_interleaved_s8s32_dot_8x3VL/a64fx.cpp",
13771402
"src/core/NEON/kernels/arm_gemm/kernels/sve_interleaved_s8s32_dot_8x3VL/generic.cpp",
13781403
"src/core/NEON/kernels/arm_gemm/kernels/sve_interleaved_s8s32_mmla_8x3VL/generic.cpp",
1404+
"src/core/NEON/kernels/arm_gemm/kernels/sve_interleaved_u8s8s32_mmla_8x3VL/generic.cpp",
13791405
"src/core/NEON/kernels/arm_gemm/kernels/sve_interleaved_u8u32_dot_8x3VL/a64fx.cpp",
13801406
"src/core/NEON/kernels/arm_gemm/kernels/sve_interleaved_u8u32_dot_8x3VL/generic.cpp",
13811407
"src/core/NEON/kernels/arm_gemm/kernels/sve_interleaved_u8u32_mmla_8x3VL/generic.cpp",

CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ cmake_minimum_required(VERSION 3.13 FATAL_ERROR)
2828
list(APPEND CMAKE_MESSAGE_CONTEXT ArmCompute)
2929
project(
3030
ArmCompute
31-
VERSION 39.0.0
31+
VERSION 40.0.0
3232
DESCRIPTION
3333
"The Arm Compute Library is a collection of low-level machine learning functions optimized for Arm® Cortex®-A CPU and Arm® Mali™ GPU architectures"
3434
LANGUAGES C CXX ASM)

CONTRIBUTING.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
Please read https://arm-software.github.io/ComputeLibrary/latest/contribution_guidelines.xhtml
1+
Please read https://artificial-intelligence.sites.arm.com/computelibrary/latest/contribution_guidelines.xhtml
22

33
Here on github we only publish a snapshot of the main development branch for each release, that's the reason why we don't accept pull requests.
44

README.md

Lines changed: 13 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
12
> **⚠ Deprecation Notice**
23
> 24.01 announcement: NCHW data format specific optimizations will gradually be removed from the code base in
34
> future releases. The implication of this is that the user is expected to translate NCHW models into NHWC in
@@ -8,7 +9,7 @@
89
<img src="https://raw.githubusercontent.com/ARM-software/ComputeLibrary/gh-pages/ACL_logo.png"/><br><br>
910
</div>
1011

11-
# Compute Library ![](https://img.shields.io/badge/latest_release-24.07-green)
12+
# Compute Library ![](https://img.shields.io/badge/latest_release-24.08-green)
1213

1314

1415
The Compute Library is a collection of low-level machine learning functions optimized for Arm® Cortex®-A, Arm® Neoverse® and Arm® Mali™ GPUs architectures.<br>
@@ -36,7 +37,7 @@ Key Features:
3637
<br>
3738

3839
## Documentation
39-
[![Documentation](https://img.shields.io/badge/documentation-24.07-green)](https://arm-software.github.io/ComputeLibrary/latest)
40+
[![Documentation](https://img.shields.io/badge/documentation-24.08-green)](https://artificial-intelligence.sites.arm.com/computelibrary/v24.08/index.xhtml)
4041

4142
> Note: The documentation includes the reference API, changelogs, build guide, contribution guide, errata, etc.
4243
@@ -49,22 +50,22 @@ All the binaries can be downloaded from [here](https://github.com/ARM-software/C
4950

5051
| Platform | Operating System | Release archive (Download) |
5152
| -------------- | ---------------- | -------------------------- |
52-
| Raspberry Pi 4 | Linux® 32bit | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.07/arm_compute-v24.07-linux-armv7a-cpu-bin.tar.gz) |
53-
| Raspberry Pi 4 | Linux® 64bit | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.07/arm_compute-v24.07-linux-aarch64-cpu-bin.tar.gz) |
54-
| Odroid N2 | Linux® 64bit | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.07/arm_compute-v24.07-linux-aarch64-cpu-bin.tar.gz) [![](https://img.shields.io/badge/build-neon+cl-yellowgreen)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.07/arm_compute-v24.07-linux-aarch64-cpu-gpu-bin.tar.gz) |
55-
| HiKey960 | Linux® 64bit | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.07/arm_compute-v24.07-linux-aarch64-cpu-bin.tar.gz) [![](https://img.shields.io/badge/build-neon+cl-yellowgreen)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.07/arm_compute-v24.07-linux-aarch64-cpu-gpu-bin.tar.gz) |
53+
| Raspberry Pi 4 | Linux® 32bit | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.08/arm_compute-v24.08-linux-armv7a-cpu-bin.tar.gz) |
54+
| Raspberry Pi 4 | Linux® 64bit | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.08/arm_compute-v24.08-linux-aarch64-cpu-bin.tar.gz) |
55+
| Odroid N2 | Linux® 64bit | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.08/arm_compute-v24.08-linux-aarch64-cpu-bin.tar.gz) [![](https://img.shields.io/badge/build-neon+cl-yellowgreen)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.08/arm_compute-v24.08-linux-aarch64-cpu-gpu-bin.tar.gz) |
56+
| HiKey960 | Linux® 64bit | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.08/arm_compute-v24.08-linux-aarch64-cpu-bin.tar.gz) [![](https://img.shields.io/badge/build-neon+cl-yellowgreen)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.08/arm_compute-v24.08-linux-aarch64-cpu-gpu-bin.tar.gz) |
5657

5758
<br>
5859

5960
| Architecture | Operating System | Release archive (Download) |
6061
| ------------ | ---------------- | -------------------------- |
61-
| armv7 | Linux® | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.07/arm_compute-v24.07-linux-armv7a-cpu-bin.tar.gz) [![](https://img.shields.io/badge/build-neon+cl-yellowgreen)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.07/arm_compute-v24.07-linux-armv7a-cpu-gpu-bin.tar.gz) |
62-
| arm64-v8a | Android™ | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.07/arm_compute-v24.07-android-aarch64-cpu-bin.tar.gz) [![](https://img.shields.io/badge/build-neon+cl-yellowgreen)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.07/arm_compute-v24.07-android-aarch64-cpu-gpu-bin.tar.gz) |
63-
| arm64-v8a | Linux® | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.07/arm_compute-v24.07-linux-aarch64-cpu-bin.tar.gz) [![](https://img.shields.io/badge/build-neon+cl-yellowgreen)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.07/arm_compute-v24.07-linux-aarch64-cpu-gpu-bin.tar.gz) |
62+
| armv7 | Linux® | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.08/arm_compute-v24.08-linux-armv7a-cpu-bin.tar.gz) [![](https://img.shields.io/badge/build-neon+cl-yellowgreen)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.08/arm_compute-v24.08-linux-armv7a-cpu-gpu-bin.tar.gz) |
63+
| arm64-v8a | Android™ | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.08/arm_compute-v24.08-android-aarch64-cpu-bin.tar.gz) [![](https://img.shields.io/badge/build-neon+cl-yellowgreen)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.08/arm_compute-v24.08-android-aarch64-cpu-gpu-bin.tar.gz) |
64+
| arm64-v8a | Linux® | [![](https://img.shields.io/badge/build-neon-orange)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.08/arm_compute-v24.08-linux-aarch64-cpu-bin.tar.gz) [![](https://img.shields.io/badge/build-neon+cl-yellowgreen)](https://github.com/ARM-software/ComputeLibrary/releases/download/v24.08/arm_compute-v24.08-linux-aarch64-cpu-gpu-bin.tar.gz) |
6465

6566
<br>
6667

67-
Please refer to the following link for more pre-built binaries: [![](https://img.shields.io/badge/v24.07-bins-yellowgreen)](https://github.com/ARM-software/ComputeLibrary/releases/tag/v24.07)
68+
Please refer to the following link for more pre-built binaries: [![](https://img.shields.io/badge/v24.08-bins-yellowgreen)](https://github.com/ARM-software/ComputeLibrary/releases/tag/v24.08)
6869

6970
Pre-build binaries are generated with the following security / good coding practices related flags:
7071
> -Wall, -Wextra, -Wformat=2, -Winit-self, -Wstrict-overflow=2, -Wswitch-default, -Woverloaded-virtual, -Wformat-security, -Wctor-dtor-privacy, -Wsign-promo, -Weffc++, -pedantic, -fstack-protector-strong
@@ -106,13 +107,13 @@ Pre-build binaries are generated with the following security / good coding pract
106107

107108
## Experimental builds
108109

109-
**⚠ Important** Bazel and CMake builds are experimental CPU only builds, please see the [documentation](https://arm-software.github.io/ComputeLibrary/latest/how_to_build.xhtml) for more details.
110+
**⚠ Important** Bazel and CMake builds are experimental CPU only builds, please see the [documentation](https://artificial-intelligence.sites.arm.com/computelibrary/v24.08/how_to_build.xhtml) for more details.
110111

111112
<br>
112113

113114
## How to contribute
114115

115-
Contributions to the Compute Library are more than welcome. If you are interested on contributing, please have a look at our [how to contribute guidelines](https://arm-software.github.io/ComputeLibrary/latest/contribution_guidelines.xhtml).
116+
Contributions to the Compute Library are more than welcome. If you are interested on contributing, please have a look at our [how to contribute guidelines](https://artificial-intelligence.sites.arm.com/computelibrary/v24.08/contribution_guidelines.xhtml).
116117

117118
### Developer Certificate of Origin (DCO)
118119
Before the Compute Library accepts your contribution, you need to certify its origin and give us your permission. To manage this process we use the Developer Certificate of Origin (DCO) V1.1 (https://developercertificate.org/)

SConscript

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -33,10 +33,10 @@ import codecs
3333
import platform
3434
import SCons
3535

36-
VERSION = "v24.07"
37-
LIBRARY_VERSION_MAJOR = 39
38-
LIBRARY_VERSION_MINOR = 0
39-
LIBRARY_VERSION_PATCH = 0
36+
VERSION = "v24.08"
37+
LIBRARY_VERSION_MAJOR = 40
38+
LIBRARY_VERSION_MINOR = 0
39+
LIBRARY_VERSION_PATCH = 0
4040
SONAME_VERSION = str(LIBRARY_VERSION_MAJOR) + "." + str(LIBRARY_VERSION_MINOR) + "." + str(LIBRARY_VERSION_PATCH)
4141

4242
Import('env')

arm_compute/core/CPP/CPPTypes.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,7 +180,7 @@ class CPUInfo final
180180
*
181181
* @return Vector length if sme2 is enabled, otherwise returns 0.
182182
*/
183-
unsigned long get_sme2_vector_length() const;
183+
uint64_t get_sme2_vector_length() const;
184184

185185
private:
186186
struct Impl;

arm_compute/core/ITensorInfo.h

Lines changed: 11 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
/*
2-
* Copyright (c) 2016-2023 Arm Limited.
2+
* Copyright (c) 2016-2024 Arm Limited.
33
*
44
* SPDX-License-Identifier: MIT
55
*
@@ -21,8 +21,8 @@
2121
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
2222
* SOFTWARE.
2323
*/
24-
#ifndef ARM_COMPUTE_ITENSORINFO_H
25-
#define ARM_COMPUTE_ITENSORINFO_H
24+
#ifndef ACL_ARM_COMPUTE_CORE_ITENSORINFO_H
25+
#define ACL_ARM_COMPUTE_CORE_ITENSORINFO_H
2626

2727
#include "arm_compute/core/Coordinates.h"
2828
#include "arm_compute/core/Strides.h"
@@ -279,6 +279,13 @@ class ITensorInfo : public misc::ICloneable<ITensorInfo>
279279
* @return Reference to this ITensorInfo object
280280
*/
281281
virtual ITensorInfo &set_are_values_constant(bool are_values_constant) = 0;
282+
/** Set the offset from the beginning of the memory allocation to the first element of the tensor.
283+
*
284+
* @param[in] offset Offset in bytes to first element in tensor.
285+
*
286+
* @return Reference to this ITensorInfo object
287+
*/
288+
virtual ITensorInfo &set_offset_first_element_in_bytes(const size_t offset) = 0;
282289
/** Valid region of the tensor. All elements in the valid region have defined values, i.e. are not undefined.
283290
*
284291
* @return The valid region.
@@ -364,4 +371,4 @@ class ITensorInfo : public misc::ICloneable<ITensorInfo>
364371
static constexpr int32_t _static_dimension = 0;
365372
};
366373
} // namespace arm_compute
367-
#endif /*ARM_COMPUTE_TENSORINFO_H */
374+
#endif // ACL_ARM_COMPUTE_CORE_ITENSORINFO_H

0 commit comments

Comments
 (0)