206 lines
6.0 KiB
Python
206 lines
6.0 KiB
Python
# Description:
|
|
# VSI OVX wrapper logic
|
|
|
|
package(
|
|
default_visibility = ["//visibility:public"],
|
|
features = ["-parse_headers"],
|
|
)
|
|
|
|
load("@bazel_tools//tools/build_defs/pkg:pkg.bzl", "pkg_tar")
|
|
|
|
filegroup(
|
|
name = "all_files",
|
|
srcs = glob(
|
|
["**/*"],
|
|
exclude = [
|
|
"**/METADATA",
|
|
"**/OWNERS",
|
|
"external/*"
|
|
],
|
|
),
|
|
)
|
|
|
|
filegroup(
|
|
name = "kernel_hdrs",
|
|
srcs = glob([
|
|
"include/kernel/cl/*.h",
|
|
"include/kernel/evis/*.h",
|
|
])
|
|
)
|
|
|
|
filegroup(
|
|
name = "kernel_srcs",
|
|
srcs = glob([
|
|
"src/kernel/cl/*.c",
|
|
"src/kernel/evis/*.c",
|
|
"src/kernel/vx/*.c",
|
|
])
|
|
)
|
|
|
|
filegroup(
|
|
name = "operation_srcs",
|
|
srcs = glob([
|
|
"src/ops/*.c",
|
|
])
|
|
)
|
|
|
|
filegroup(
|
|
name = "operation_hdrs",
|
|
srcs = glob([
|
|
"include/ops/*.h",
|
|
])
|
|
)
|
|
|
|
filegroup(
|
|
name = "custom_hdrs",
|
|
srcs = glob([
|
|
"include/custom/ops/*.h",
|
|
]) + [
|
|
#custom
|
|
"include/custom/custom_node_type.def",
|
|
"include/custom/custom_ops.def",
|
|
"include/custom/vsi_nn_custom_node_type.h",
|
|
]
|
|
)
|
|
|
|
filegroup(
|
|
name = "custom_srcs",
|
|
srcs = glob([
|
|
"src/custom/ops/*.c",
|
|
"src/custom/ops/kernel/evis/*.c",
|
|
"src/custom/ops/kernel/cl/*.c",
|
|
"src/custom/ops/kernel/cpu/*.c",
|
|
])
|
|
)
|
|
|
|
cc_library(
|
|
name = "ovxlibimpl",
|
|
copts = [
|
|
"-Werror", "-Wmisleading-indentation",
|
|
"-fvisibility=hidden", '-DOVXLIB_API=__attribute__((visibility(\\"default\\")))',
|
|
],
|
|
linkopts = ["-ldl", "-lm"],
|
|
alwayslink=True,
|
|
linkstatic = True,
|
|
includes = [
|
|
"include",
|
|
],
|
|
hdrs = [
|
|
"include/vsi_nn_pub.h",
|
|
"include/vsi_nn_ops.h",
|
|
"include/vsi_nn_log.h",
|
|
"include/vsi_nn_context.h",
|
|
"include/vsi_nn_node_attr_template.h",
|
|
"include/vsi_nn_tensor.h",
|
|
"include/vsi_nn_prv.h",
|
|
"include/vsi_nn_types.h",
|
|
"include/vsi_nn_node.h",
|
|
"include/vsi_nn_node_type.h",
|
|
"include/vsi_nn_client_op.h",
|
|
"include/vsi_nn_graph.h",
|
|
"include/vsi_nn_test.h",
|
|
"include/vsi_nn_tensor_util.h",
|
|
"include/vsi_nn_version.h",
|
|
"include/vsi_nn_compatibility.h",
|
|
"include/vsi_nn_assert.h",
|
|
"include/vsi_nn_feature.h",
|
|
"include/vsi_nn_rnn.h",
|
|
"include/vsi_nn_rnn_helper.h",
|
|
"include/vsi_nn_rnn_prv.h",
|
|
"include/vsi_nn_internal_node.h",
|
|
"include/vsi_nn_daemon.h",
|
|
"include/vsi_nn_pre_post_process.h",
|
|
"include/vsi_nn_graph_optimization.h",
|
|
"include/utils/vsi_nn_link_list.h",
|
|
"include/utils/vsi_nn_math.h",
|
|
"include/utils/vsi_nn_util.h",
|
|
"include/utils/vsi_nn_code_generator.h",
|
|
"include/utils/vsi_nn_binary_tree.h",
|
|
"include/utils/vsi_nn_map.h",
|
|
"include/utils/vsi_nn_hashmap.h",
|
|
"include/utils/vsi_nn_limits.h",
|
|
"include/utils/vsi_nn_dtype_util.h",
|
|
"include/utils/vsi_nn_dtype_util_prv.h",
|
|
"include/utils/vsi_nn_vdata.h",
|
|
"include/utils/vsi_nn_tensor_op.h",
|
|
"include/utils/vsi_nn_shape_util.h",
|
|
"include/utils/vsi_nn_constraint_check.h",
|
|
"include/quantization/vsi_nn_asymmetric_affine.h",
|
|
"include/quantization/vsi_nn_dynamic_fixed_point.h",
|
|
"include/quantization/vsi_nn_perchannel_symmetric_affine.h",
|
|
"include/interface/ops.def",
|
|
"include/kernel/vsi_nn_kernel.h",
|
|
"include/kernel/vsi_nn_gpu.h",
|
|
"include/kernel/vsi_nn_gpu_config.h",
|
|
"include/kernel/vsi_nn_kernel_eltwise.h",
|
|
"include/kernel/vsi_nn_kernel_node.h",
|
|
"include/kernel/vsi_nn_kernel_gpu_shape_optimize.h",
|
|
"include/kernel/vsi_nn_kernel_lut.h",
|
|
"include/vsi_nn_error.h",
|
|
|
|
# libnnext
|
|
"include/libnnext/vsi_nn_vxkernel.h",
|
|
"include/libnnext/vx_lib_nnext.h",
|
|
"include/libnnext/vsi_nn_libnnext_resource.h",
|
|
|
|
#internal
|
|
"include/internal/internal_ops.def",
|
|
"include/vsi_nn_feature_config.h"
|
|
] + [":kernel_hdrs"]
|
|
+ [":operation_hdrs"]
|
|
+ [":custom_hdrs"]
|
|
+ [
|
|
"include/vsi_nn_platform.h",
|
|
],
|
|
srcs = [
|
|
"src/vsi_nn_graph.c",
|
|
"src/vsi_nn_ops.c",
|
|
"src/vsi_nn_context.c",
|
|
"src/vsi_nn_node.c",
|
|
"src/vsi_nn_tensor.c",
|
|
"src/vsi_nn_client_op.c",
|
|
"src/vsi_nn_node_attr_template.c",
|
|
"src/vsi_nn_version.c",
|
|
"src/vsi_nn_rnn.c",
|
|
"src/vsi_nn_rnn_helper.c",
|
|
"src/vsi_nn_log.c",
|
|
"src/vsi_nn_internal_node.c",
|
|
"src/vsi_nn_daemon.c",
|
|
"src/vsi_nn_graph_optimization.c",
|
|
"src/vsi_nn_pre_post_process.c",
|
|
"src/utils/vsi_nn_link_list.c",
|
|
"src/utils/vsi_nn_util.c",
|
|
"src/utils/vsi_nn_math.c",
|
|
"src/utils/vsi_nn_code_generator.c",
|
|
"src/utils/vsi_nn_binary_tree.c",
|
|
"src/utils/vsi_nn_map.c",
|
|
"src/utils/vsi_nn_hashmap.c",
|
|
"src/utils/vsi_nn_limits.c",
|
|
"src/utils/vsi_nn_dtype_util.c",
|
|
"src/utils/vsi_nn_vdata.c",
|
|
"src/utils/vsi_nn_tensor_op.c",
|
|
"src/utils/vsi_nn_shape_util.c",
|
|
"src/utils/vsi_nn_dtype.c",
|
|
"src/utils/vsi_nn_constraint_check.c",
|
|
"src/quantization/vsi_nn_asymmetric_affine.c",
|
|
"src/quantization/vsi_nn_dynamic_fixed_point.c",
|
|
"src/quantization/vsi_nn_perchannel_symmetric_affine.c",
|
|
"src/kernel/vsi_nn_kernel.c",
|
|
"src/kernel/vsi_nn_kernel_util.c",
|
|
"src/kernel/vsi_nn_kernel_backend.c",
|
|
"src/kernel/vsi_nn_kernel_eltwise.c",
|
|
"src/kernel/vsi_nn_kernel_selector.c",
|
|
"src/kernel/vsi_nn_kernel_node.c",
|
|
"src/kernel/vsi_nn_kernel_param.c",
|
|
"src/kernel/vsi_nn_kernel_lut.c",
|
|
"src/kernel/vsi_nn_gpu.c",
|
|
"src/kernel/vsi_nn_kernel_gpu_shape_optimize.c",
|
|
"src/libnnext/vsi_nn_libnnext_resource.c",
|
|
"src/libnnext/vsi_nn_vxkernel.c",
|
|
] + [":kernel_srcs"]
|
|
+ [":operation_srcs"]
|
|
+ [":custom_srcs"],
|
|
deps = ["//prebuilt-sdk:VIV_SDK_LIB"]
|
|
)
|
|
|