diff options
| author | Sheri Zhang <sheri.zhang@arm.com> | 2021-02-08 17:43:04 +0000 |
|---|---|---|
| committer | Sheri Zhang <sheri.zhang@arm.com> | 2021-02-10 10:58:45 +0000 |
| commit | 79144a642b33ff1ac40a44aaa1881261d12e6376 (patch) | |
| tree | 81e25f9f99c03436fa3563b94881ea879e9b15c1 /scripts/arm_compute_library_nn_driver.go | |
| parent | 8c23ba1c5041467f314a10c1da9147e41d056139 (diff) | |
| download | ComputeLibrary-79144a642b33ff1ac40a44aaa1881261d12e6376.tar.gz | |
Decouple CpuPoolingKernel data type and data layout
1. Decouple data layout for CpuPoolingKernel: NCHW & NHWC
2. Decouple data type for CpuPoolingKernel NHWC
Partially solves: COMPMID-3999
Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: I3c6535eebdddeb467b7c68a7287a16959b5b9695
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5039
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'scripts/arm_compute_library_nn_driver.go')
| -rw-r--r-- | scripts/arm_compute_library_nn_driver.go | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/scripts/arm_compute_library_nn_driver.go b/scripts/arm_compute_library_nn_driver.go index 8ff4f69495..d573c6a14a 100644 --- a/scripts/arm_compute_library_nn_driver.go +++ b/scripts/arm_compute_library_nn_driver.go @@ -48,6 +48,17 @@ func globalFlags(ctx android.BaseContext) []string { } } + data_layouts := strings.Split(ctx.AConfig().GetenvWithDefault("COMPUTE_LIB_DATA_LAYOUT", "ALL"), ",") + + for _, x := range data_layouts { + if strings.ToUpper(x) == "ALL" || strings.ToUpper(x) == "NHWC" { + cppflags = append(cppflags, "-DENABLE_NHWC_KERNELS") + } + if strings.ToUpper(x) == "ALL" || strings.ToUpper(x) == "NCHW" { + cppflags = append(cppflags, "-DENABLE_NCHW_KERNELS") + } + } + return cppflags } |