diff --git a/mace/codegen/BUILD b/mace/codegen/BUILD index c6d0f9ff105f82828987d1781e758113d1266094..811b420ed27351422bae9f0b35981846dbdc5da3 100644 --- a/mace/codegen/BUILD +++ b/mace/codegen/BUILD @@ -21,6 +21,19 @@ cc_library( linkstatic = 1, ) +cc_library( + name = "generated_opencl_prod", + srcs = ["opencl/opencl_compiled_program.cc"], + linkstatic = 1, +) + +cc_library( + name = "generated_tuning_params", + srcs = ["tuning/tuning_params.cc"], + linkstatic = 1, +) + + cc_library( name = "generated_version", srcs = ["version/version.cc"], diff --git a/mace/python/tools/str2vec_maps.cc.tmpl b/mace/python/tools/str2vec_maps.cc.tmpl new file mode 100644 index 0000000000000000000000000000000000000000..354af0aa38e18eebb344cb6493de2d17e0946a92 --- /dev/null +++ b/mace/python/tools/str2vec_maps.cc.tmpl @@ -0,0 +1,27 @@ +// +// Copyright (c) 2017 XiaoMi All rights reserved. +// + +// This is a generated file, DO NOT EDIT + +#include +#include +#include + +namespace mace { + +extern const std::map> {{variable_name}}= +{ + {% for key, value in maps.iteritems() %} + { + "{{key}}", + { + {%- for ele in value -%} + {{ele}}, + {%- endfor -%} + } + }, // {{key}} +{% endfor %} +}; + +} // namespace