diff options
author | Jakub Sujak <jakub.sujak@arm.com> | 2023-06-16 09:52:50 +0100 |
---|---|---|
committer | Jakub Sujak <jakub.sujak@arm.com> | 2023-06-26 13:07:05 +0000 |
commit | 8c49f16e5909a9bd5dc6e68638d2e2d8acc2fc66 (patch) | |
tree | 7e6c13c7f4522ea2db1ccdafe7c2858632ee4532 /SConstruct | |
parent | 7d9a78ebfb3553b95421a0da5e2686a3923748db (diff) | |
download | ComputeLibrary-8c49f16e5909a9bd5dc6e68638d2e2d8acc2fc66.tar.gz |
Add helpers to set CKW tensor components as OpenCL kernel arguments
* Define ckw::TensorStorage. The tensor storage represents the type of tensor memory object.
* Add helper functions for setting the CKW TensorComponent and TensorStorage as OpenCL kernel arguments.
* Refactor CL Image2D method for simpler image object creation.
Resolves: COMPMID-5784
Change-Id: I2d37d06783c1dc55f3b5692b44eb49b151f2401c
Signed-off-by: Jakub Sujak <jakub.sujak@arm.com>
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/9807
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'SConstruct')
-rw-r--r-- | SConstruct | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/SConstruct b/SConstruct index 03b0f918e2..419fa33558 100644 --- a/SConstruct +++ b/SConstruct @@ -126,7 +126,7 @@ vars.AddVariables( ├── datasets ├── fixtures └── Neon\n""", "", PathVariable.PathAccept), - BoolVariable("experimental_dynamic_fusion", "Build the experimental dynamic fusion files", False), + BoolVariable("experimental_dynamic_fusion", "Build the experimental dynamic fusion files. This option also enables opencl=1 and ckw=1 on which it has a direct dependency.", False), BoolVariable("fixed_format_kernels", "Enable fixed format kernels for GEMM", False), BoolVariable("mapfile", "Generate a map file", False), ListVariable("custom_options", "Custom options that can be used to turn on/off features", "none", ["disable_mmla_fp"]), @@ -215,6 +215,11 @@ if env['os'] == 'bare_metal': print("ERROR: OpenMP and C++11 threads not supported in bare_metal. Use cppthreads=0 openmp=0") Exit(1) +if env['experimental_dynamic_fusion']: + # Dynamic Fusion on GPU has a direct dependency on OpenCL and Compute Kernel Writer + env['opencl'] = 1 + env['ckw'] = 1 + if env['opencl'] and env['embed_kernels'] and env['compress_kernels'] and env['os'] not in ['android']: print("Compressed kernels are supported only for android builds") Exit(1) |