File tree Expand file tree Collapse file tree 3 files changed +6
-14
lines changed
backends/xnnpack/third-party Expand file tree Collapse file tree 3 files changed +6
-14
lines changed Original file line number Diff line number Diff line change @@ -42,7 +42,7 @@ def define_xnnpack():
4242 "XNNPACK/src/mutex.c" ,
4343 "XNNPACK/src/normalization.c" ,
4444 "XNNPACK/src/operator-utils.c" ,
45- "XNNPACK/src/packing.cc" ,
45+ "XNNPACK/src/reference/ packing.cc" ,
4646 ],
4747 headers = get_xnnpack_headers (),
4848 header_namespace = "" ,
@@ -67,7 +67,7 @@ def define_xnnpack():
6767 # @lint-ignore BUCKLINT: native and fb_native are explicitly forbidden in fbcode.
6868 native .cxx_library (
6969 name = "subgraph" ,
70- srcs = SUBGRAPH_SRCS ,
70+ srcs = SUBGRAPH_SRCS + [ "XNNPACK/src/datatype.c" ] ,
7171 compiler_flags = [
7272 "-Wno-error=missing-braces" , # required since the SGX toolchain does not have this by default
7373 ],
@@ -1076,6 +1076,8 @@ def define_xnnpack():
10761076 "XNNPACK/src/configs/hardware-config.c" ,
10771077 "XNNPACK/src/microparams-init.c" ,
10781078 "XNNPACK/src/microkernel-utils.c" ,
1079+ "XNNPACK/src/reference/binary-elementwise.cc" ,
1080+ "XNNPACK/src/reference/unary-elementwise.cc" ,
10791081 ],
10801082 headers = get_xnnpack_headers (),
10811083 exported_headers = {
Original file line number Diff line number Diff line change @@ -17,24 +17,14 @@ def prod_srcs_for_arch_wrapper(arch):
1717 return define_xnnpack_build_src (prod_srcs )
1818
1919def get_xnnpack_headers ():
20- # XNNPACK Headers in the path containing xnnpack/ or configs/
21- # do not contain the src/ path. However headers not in xnnpack/ or
22- # configs/ are prepend with the src/ path. This function helps us
23- # to correctly parse all the header files to the correct name
2420 src_headers = subdir_glob ([
2521 ("XNNPACK/src" , "**/*.h" ),
2622 ])
27- fixed_headers = {}
28- for k , v in src_headers .items ():
29- new_key = k
30- if not k .startswith ("xnnpack" ) and not k .startswith ("configs" ):
31- new_key = "src/{}" .format (k )
32- fixed_headers [new_key ] = v
3323 include_headers = subdir_glob ([
3424 ("XNNPACK/include" , "*.h" ),
3525 ])
3626
37- return fixed_headers | include_headers
27+ return src_headers | include_headers
3828
3929OPERATOR_SRCS = define_xnnpack_build_src (_OPERATOR_SRCS )
4030SUBGRAPH_SRCS = define_xnnpack_build_src (_SUBGRAPH_SRCS )
You can’t perform that action at this time.
0 commit comments