Internal change
PiperOrigin-RevId: 515094293
This commit is contained in:
parent
ddc535e705
commit
09f63cbbe0
|
@ -22,6 +22,7 @@ load(
|
||||||
load("//mediapipe/framework:mediapipe_cc_test.bzl", "mediapipe_cc_test")
|
load("//mediapipe/framework:mediapipe_cc_test.bzl", "mediapipe_cc_test")
|
||||||
load("//mediapipe/framework:encode_binary_proto.bzl", "encode_binary_proto")
|
load("//mediapipe/framework:encode_binary_proto.bzl", "encode_binary_proto")
|
||||||
load("@org_tensorflow//tensorflow/lite/core/shims:cc_library_with_tflite.bzl", "cc_library_with_tflite")
|
load("@org_tensorflow//tensorflow/lite/core/shims:cc_library_with_tflite.bzl", "cc_library_with_tflite")
|
||||||
|
load("//mediapipe/framework:more_selects.bzl", "more_selects")
|
||||||
|
|
||||||
licenses(["notice"])
|
licenses(["notice"])
|
||||||
|
|
||||||
|
@ -35,6 +36,27 @@ exports_files(
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
config_setting(
|
||||||
|
name = "disable_gpu",
|
||||||
|
define_values = {
|
||||||
|
"MEDIAPIPE_DISABLE_GPU": "1",
|
||||||
|
},
|
||||||
|
visibility = ["//visibility:public"],
|
||||||
|
)
|
||||||
|
|
||||||
|
more_selects.config_setting_negation(
|
||||||
|
name = "not_disable_gpu",
|
||||||
|
negate = ":disable_gpu",
|
||||||
|
)
|
||||||
|
|
||||||
|
selects.config_setting_group(
|
||||||
|
name = "platform_ios_with_gpu",
|
||||||
|
match_all = [
|
||||||
|
":not_disable_gpu",
|
||||||
|
"//mediapipe:ios",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
selects.config_setting_group(
|
selects.config_setting_group(
|
||||||
name = "compute_shader_unavailable",
|
name = "compute_shader_unavailable",
|
||||||
match_any = [
|
match_any = [
|
||||||
|
@ -585,7 +607,7 @@ cc_library(
|
||||||
":inference_calculator_cpu",
|
":inference_calculator_cpu",
|
||||||
] + select({
|
] + select({
|
||||||
"//conditions:default": [":inference_calculator_gl_if_compute_shader_available"],
|
"//conditions:default": [":inference_calculator_gl_if_compute_shader_available"],
|
||||||
"//mediapipe:ios": [":inference_calculator_metal"],
|
":platform_ios_with_gpu": [":inference_calculator_metal"],
|
||||||
}),
|
}),
|
||||||
alwayslink = 1,
|
alwayslink = 1,
|
||||||
)
|
)
|
||||||
|
|
|
@ -272,6 +272,14 @@ selects.config_setting_group(
|
||||||
],
|
],
|
||||||
)
|
)
|
||||||
|
|
||||||
|
selects.config_setting_group(
|
||||||
|
name = "platform_ios_without_gpu",
|
||||||
|
match_all = [
|
||||||
|
":disable_gpu",
|
||||||
|
"//mediapipe:ios",
|
||||||
|
],
|
||||||
|
)
|
||||||
|
|
||||||
selects.config_setting_group(
|
selects.config_setting_group(
|
||||||
name = "platform_macos_with_gpu",
|
name = "platform_macos_with_gpu",
|
||||||
match_all = [
|
match_all = [
|
||||||
|
@ -310,6 +318,9 @@ cc_library(
|
||||||
":gl_texture_view",
|
":gl_texture_view",
|
||||||
":gl_texture_buffer",
|
":gl_texture_buffer",
|
||||||
],
|
],
|
||||||
|
":platform_ios_without_gpu": [
|
||||||
|
"//mediapipe/objc:util",
|
||||||
|
],
|
||||||
":disable_gpu": [],
|
":disable_gpu": [],
|
||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
|
|
Loading…
Reference in New Issue
Block a user