This commit is contained in:
Esun Kim 2026-01-24 10:34:15 -08:00 committed by GitHub
parent dec1a10898
commit db0e3d9bf2
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
5 changed files with 44 additions and 1 deletions

View file

@ -62,6 +62,11 @@ py_tflm_signal_library(
],
)
TARGET_COMPATIBLE_WITH = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
})
py_test(
name = "delay_op_test",
size = "small",
@ -71,6 +76,7 @@ py_test(
"nomsan",
"noubsan",
],
target_compatible_with = TARGET_COMPATIBLE_WITH,
deps = [
":delay_op",
requirement("numpy"),
@ -102,6 +108,7 @@ py_test(
"nomsan",
"noubsan",
],
target_compatible_with = TARGET_COMPATIBLE_WITH,
deps = [
":energy_op",
requirement("numpy"),
@ -133,6 +140,7 @@ py_test(
"nomsan",
"noubsan",
],
target_compatible_with = TARGET_COMPATIBLE_WITH,
deps = [
":fft_ops",
requirement("numpy"),
@ -170,6 +178,7 @@ py_test(
"nomsan",
"noubsan",
],
target_compatible_with = TARGET_COMPATIBLE_WITH,
deps = [
":filter_bank_ops",
requirement("numpy"),
@ -201,6 +210,7 @@ py_test(
"nomsan",
"noubsan",
],
target_compatible_with = TARGET_COMPATIBLE_WITH,
deps = [
":framer_op",
requirement("numpy"),
@ -229,6 +239,7 @@ py_test(
"nomsan",
"noubsan",
],
target_compatible_with = TARGET_COMPATIBLE_WITH,
deps = [
":overlap_add_op",
requirement("absl_py"),
@ -261,6 +272,7 @@ py_test(
"nomsan",
"noubsan",
],
target_compatible_with = TARGET_COMPATIBLE_WITH,
deps = [
":pcan_op",
requirement("numpy"),
@ -292,6 +304,7 @@ py_test(
"nomsan",
"noubsan",
],
target_compatible_with = TARGET_COMPATIBLE_WITH,
deps = [
":stacker_op",
requirement("numpy"),
@ -322,6 +335,7 @@ py_test(
"nomsan",
"noubsan",
],
target_compatible_with = TARGET_COMPATIBLE_WITH,
deps = [
":window_op",
requirement("numpy"),

View file

@ -35,6 +35,7 @@ def py_tflm_signal_library(
if srcs:
binary_path_end_pos = srcs[0].rfind("/")
binary_path = srcs[0][0:binary_path_end_pos]
binary_name = binary_path + "/_" + cc_op_kernels[0][1:] + ".so"
if cc_op_defs:
binary_name = "ops/_" + name + ".so"

View file

@ -11,6 +11,11 @@ package(
licenses = ["notice"],
)
TARGET_COMPATIBLE_WITH = select({
"@bazel_tools//src/conditions:windows": ["@platforms//:incompatible"],
"//conditions:default": [],
})
py_test(
name = "freq_to_mel_test",
srcs = ["freq_to_mel_test.py"],
@ -22,6 +27,7 @@ py_test(
"nomsan",
"noubsan",
],
target_compatible_with = TARGET_COMPATIBLE_WITH,
)
py_library(
@ -73,5 +79,6 @@ py_test(
"nomsan",
"noubsan",
],
target_compatible_with = TARGET_COMPATIBLE_WITH,
visibility = ["//visibility:public"],
)