1# Any targets that should be shared between fbcode and xplat must be defined in 2# targets.bzl. This file can contain fbcode-only targets. 3 4load("@fbsource//xplat/executorch/build:runtime_wrapper.bzl", "runtime") 5load(":targets.bzl", "define_common_targets") 6 7oncall("executorch") 8 9define_common_targets() 10 11runtime.python_test( 12 name = "test_sdpa_with_kv_cache", 13 srcs = [ 14 "test_sdpa_with_kv_cache.py", 15 ], 16 preload_deps = [ 17 ":custom_ops_aot_lib_mkl_noomp", 18 ":custom_ops_aot_py", 19 ], 20 deps = [ 21 "//caffe2:torch", 22 ], 23) 24 25runtime.python_test( 26 name = "test_update_quantized_cache", 27 srcs = [ 28 "test_update_quantized_cache.py", 29 ], 30 preload_deps = [ 31 ":custom_ops_aot_lib", 32 ], 33 deps = [ 34 "//caffe2:torch", 35 ], 36) 37 38runtime.python_test( 39 name = "test_preprocess_custom_ops", 40 srcs = [ 41 "test_preprocess_custom_ops.py", 42 ], 43 preload_deps = [ 44 ":preprocess_custom_ops_py", 45 ], 46 deps = [ 47 "//caffe2:torch", 48 ], 49) 50