@@ -8,11 +8,22 @@ set -eux
8
8
9
9
# TODO: expand this to //...
10
10
# TODO: can't query cadence & vulkan backends
11
+ # TODO: can't query //kernels/prim_ops because of a cpp_unittest and
12
+ # broken code in shim to read oss.folly_cxx_tests. Sending fix but it
13
+ # needs to propagate and we need a submodule update.
11
14
buck2 query " //backends/apple/... + //backends/example/... + \
12
15
//backends/mediatek/... + //backends/test/... + //backends/transforms/... + \
13
- //backends/xnnpack/... + //configurations/... + //kernels/portable/cpu/... + \
14
- //runtime/... + //schema/... + //test/... + //util/..."
16
+ //backends/xnnpack/... + //configurations/... + //kernels/aten/... + \
17
+ //kernels/optimized/... + //kernels/portable/... + //kernels/quantized/... + \
18
+ //kernels/test/... + //runtime/... + //schema/... + //test/... + //util/..."
15
19
20
+ UNBUILDABLE_OPTIMIZED_OPS_REGEX=" gelu|fft_r2c|log_softmax"
21
+ BUILDABLE_OPTIMIZED_OPS=$( buck2 query //kernels/optimized/cpu/... | grep -E -v $UNBUILDABLE_OPTIMIZED_OPS_REGEX )
22
+
23
+ BUILDABLE_KERNELS_PRIM_OPS_TARGETS=$( buck2 query //kernels/prim_ops/... | grep -v prim_ops_test_py)
16
24
# TODO: expand the covered scope of Buck targets.
17
- buck2 build //runtime/core/portable_type/...
18
- buck2 test //runtime/core/portable_type/...
25
+ # //runtime/kernel/... is failing because //third-party:torchgen_files's shell script can't find python on PATH.
26
+ # //runtime/test/... requires Python torch, which we don't have in our OSS buck setup.
27
+ buck2 test $BUILDABLE_OPTIMIZED_OPS //kernels/portable/... \
28
+ $BUILDABLE_KERNELS_PRIM_OPS_TARGETS //runtime/backend/... //runtime/core/... \
29
+ //runtime/executor: //runtime/kernel/... //runtime/platform/...
0 commit comments