aboutsummaryrefslogtreecommitdiff
path: root/SConstruct
diff options
context:
space:
mode:
authorJakub Sujak <jakub.sujak@arm.com>2023-06-16 09:52:50 +0100
committerJakub Sujak <jakub.sujak@arm.com>2023-06-26 13:07:05 +0000
commit8c49f16e5909a9bd5dc6e68638d2e2d8acc2fc66 (patch)
tree7e6c13c7f4522ea2db1ccdafe7c2858632ee4532 /SConstruct
parent7d9a78ebfb3553b95421a0da5e2686a3923748db (diff)
downloadComputeLibrary-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--SConstruct7
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)