diff options
Diffstat (limited to 'src/dynamic_fusion/sketch/gpu/GpuLogicalKernel.cpp')
| -rw-r--r-- | src/dynamic_fusion/sketch/gpu/GpuLogicalKernel.cpp | 16 |
1 files changed, 14 insertions, 2 deletions
diff --git a/src/dynamic_fusion/sketch/gpu/GpuLogicalKernel.cpp b/src/dynamic_fusion/sketch/gpu/GpuLogicalKernel.cpp index 00f625de28..c99984fc0e 100644 --- a/src/dynamic_fusion/sketch/gpu/GpuLogicalKernel.cpp +++ b/src/dynamic_fusion/sketch/gpu/GpuLogicalKernel.cpp @@ -1,5 +1,5 @@ /* - * Copyright (c) 2022 Arm Limited. + * Copyright (c) 2022-2023 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -29,7 +29,11 @@ #include "src/dynamic_fusion/sketch/gpu/GpuComponentServices.h" #include "src/dynamic_fusion/sketch/gpu/components/IGpuKernelComponent.h" #include "src/dynamic_fusion/sketch/gpu/components/cl/ClComponentStore.h" +#ifndef ACL_INTERNAL_TEST_CKW_IN_DF #include "src/dynamic_fusion/sketch/gpu/template_writer/cl/ClTemplateWriter.h" +#else // ACL_INTERNAL_TEST_CKW_IN_DF +#include "src/dynamic_fusion/sketch/gpu/ckw_driver/GpuCkwDriver.h" +#endif // ACL_INTERNAL_TEST_CKW_IN_DF namespace arm_compute { @@ -46,11 +50,19 @@ GpuLogicalKernel::GpuLogicalKernel(GpuComponentServices *services, const GpuKern GpuKernelSourceCode GpuLogicalKernel::write_kernel_code() { GpuKernelSourceCode code; - ClTemplateWriter writer{ _comp_group }; +#ifndef ACL_INTERNAL_TEST_CKW_IN_DF + ClTemplateWriter writer { _comp_group }; +#else // ACL_INTERNAL_TEST_CKW_IN_DF + GpuCkwDriver writer { _comp_group }; +#endif // ACL_INTERNAL_TEST_CKW_IN_DF code.name(writer.get_name()); code.code(writer.get_code()); +#ifndef ACL_INTERNAL_TEST_CKW_IN_DF code.arguments(writer.get_tensors()); +#else // ACL_INTERNAL_TEST_CKW_IN_DF + code.arguments(writer.get_kernel_arguments()); +#endif // ACL_INTERNAL_TEST_CKW_IN_DF code.build_options(writer.get_build_options()); code.config_id(writer.get_config_id()); code.window(writer.get_window()); |