12
12
"CXX",
13
13
)
14
14
load("@fbsource//xplat/executorch/codegen:codegen.bzl", "executorch_generated_lib")
15
+ load("@fbcode_macros//build_defs:cpp_python_extension.bzl", "cpp_python_extension")
15
16
16
17
oncall("odai_jarvis")
17
18
@@ -275,7 +276,6 @@ python_library(
275
276
"//executorch/exir/passes:spec_prop_pass",
276
277
],
277
278
)
278
-
279
279
python_library(
280
280
name = "decompose_ops",
281
281
srcs = [
@@ -293,6 +293,14 @@ python_library(
293
293
],
294
294
)
295
295
296
+ python_library(
297
+ name = "typing_stubs",
298
+ srcs = [
299
+ "typing_stubs.py",
300
+ ],
301
+ typing = True,
302
+ )
303
+
296
304
297
305
python_unittest(
298
306
name = "test_graph_builder",
@@ -321,6 +329,7 @@ python_unittest(
321
329
deps = [
322
330
"fbsource//third-party/pypi/parameterized:parameterized",
323
331
":compiler",
332
+ ":typing_stubs",
324
333
":replace_ops",
325
334
"//caffe2:torch",
326
335
"//executorch/backends/cadence/aot:compiler",
@@ -344,6 +353,7 @@ python_unittest(
344
353
":compiler",
345
354
":decompose_ops",
346
355
"//caffe2:torch",
356
+ ":typing_stubs",
347
357
"//executorch/backends/cadence/aot:compiler",
348
358
"//executorch/backends/cadence/aot:graph_builder",
349
359
"//executorch/backends/cadence/aot:pass_utils",
@@ -363,6 +373,7 @@ python_unittest(
363
373
deps = [
364
374
"fbsource//third-party/pypi/parameterized:parameterized",
365
375
":compiler",
376
+ ":typing_stubs",
366
377
"//caffe2:torch",
367
378
"//executorch/backends/cadence/aot:compiler",
368
379
"//executorch/backends/cadence/aot:fuse_ops",
@@ -384,6 +395,7 @@ python_unittest(
384
395
deps = [
385
396
"fbsource//third-party/pypi/parameterized:parameterized",
386
397
"fbsource//third-party/pypi/pyre-extensions:pyre-extensions",
398
+ ":typing_stubs",
387
399
":compiler",
388
400
"//caffe2:torch",
389
401
"//executorch/backends/cadence/aot:compiler",
@@ -404,6 +416,7 @@ python_unittest(
404
416
supports_static_listing = False,
405
417
typing = True,
406
418
deps = [
419
+ ":typing_stubs",
407
420
"fbsource//third-party/pypi/parameterized:parameterized",
408
421
"//caffe2:torch",
409
422
"//executorch/backends/cadence/aot:compiler",
@@ -477,6 +490,7 @@ python_unittest(
477
490
deps = [
478
491
":compiler",
479
492
":memory_planning",
493
+ ":typing_stubs",
480
494
":ops_registrations",
481
495
":pass_utils",
482
496
"//caffe2:torch",
0 commit comments