diff --git a/mediapipe/calculators/tensor/BUILD b/mediapipe/calculators/tensor/BUILD index b4461b46c..7a29c3af8 100644 --- a/mediapipe/calculators/tensor/BUILD +++ b/mediapipe/calculators/tensor/BUILD @@ -22,6 +22,7 @@ load( load("//mediapipe/framework:mediapipe_cc_test.bzl", "mediapipe_cc_test") 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("//mediapipe/framework:more_selects.bzl", "more_selects") 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( name = "compute_shader_unavailable", match_any = [ @@ -585,7 +607,7 @@ cc_library( ":inference_calculator_cpu", ] + select({ "//conditions:default": [":inference_calculator_gl_if_compute_shader_available"], - "//mediapipe:ios": [":inference_calculator_metal"], + ":platform_ios_with_gpu": [":inference_calculator_metal"], }), alwayslink = 1, ) diff --git a/mediapipe/gpu/BUILD b/mediapipe/gpu/BUILD index fe1cab29d..83948226a 100644 --- a/mediapipe/gpu/BUILD +++ b/mediapipe/gpu/BUILD @@ -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( name = "platform_macos_with_gpu", match_all = [ @@ -310,6 +318,9 @@ cc_library( ":gl_texture_view", ":gl_texture_buffer", ], + ":platform_ios_without_gpu": [ + "//mediapipe/objc:util", + ], ":disable_gpu": [], }), )