-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathBUILD
119 lines (119 loc) · 5.2 KB
/
BUILD
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
cc_binary(
name = "demo-pjrt",
srcs = [
"src/demo-pjrt.cpp",
"@xla//xla:autotune_results.pb.cc",
"@xla//xla:autotuning.pb.cc",
"@xla//xla/service:buffer_assignment.pb.cc",
"@xla//xla/service:hlo.pb.cc",
"@xla//xla/service/gpu:backend_configs.pb.cc",
"@xla//xla/stream_executor:device_description.pb.cc",
],
copts = [
"-Werror=unused-variable",
"-Werror=unused-but-set-variable",
"-Werror=return-type",
"-Werror=unused-result",
"-Wno-error=stringop-truncation",
],
linkstatic = True,
tags = [
"demo",
],
deps = [
# "@enzyme//:EnzymeMLIR",
# "@llvm-project//mlir:AffineDialect",
"@llvm-project//mlir:AllPassesAndDialects",
"@llvm-project//mlir:ArithDialect",
# "@llvm-project//mlir:AsyncDialect",
# "@llvm-project//mlir:ComplexDialect",
# "@llvm-project//mlir:ControlFlowDialect",
# "@llvm-project//mlir:ConversionPasses",
# "@llvm-project//mlir:DLTIDialect",
"@llvm-project//mlir:FuncDialect",
"@llvm-project//mlir:FromLLVMIRTranslation",
# "@llvm-project//mlir:GPUDialect",
# "@llvm-project//mlir:LinalgDialect",
"@llvm-project//mlir:LLVMDialect",
"@llvm-project//mlir:MathDialect",
# "@llvm-project//mlir:MemRefDialect",
# "@llvm-project//mlir:MlirOptLib",
# "@llvm-project//mlir:NVVMDialect",
# "@llvm-project//mlir:OpenMPDialect",
# "@llvm-project//mlir:Pass",
# "@llvm-project//mlir:SCFDialect",
# "@llvm-project//mlir:TransformDialect",
# "@llvm-project//mlir:Transforms",
# "@llvm-project//mlir:LLVMIRToLLVMTranslation",
# "@llvm-project//mlir:LLVMIRToNVVMTranslation",
"@llvm-project//mlir:LLVMIRTransforms",
"@llvm-project//llvm:IRReader",
"@llvm-project//llvm:Support",
"@llvm-project//llvm:AArch64AsmParser",
"@llvm-project//llvm:AArch64CodeGen",
"@llvm-project//llvm:X86AsmParser",
"@llvm-project//llvm:X86CodeGen",
# "@enzyme_ad//src/enzyme_ad/jax:TransformOps",
# "@enzyme_ad//src/enzyme_ad/jax:XLADerivatives",
# "@enzyme_ad//src/enzyme_ad/jax:gpu",
"@xla//xla/ffi/api:ffi",
"@xla//xla/ffi:ffi_api",
"@stablehlo//:chlo_ops",
"@xla//xla/pjrt:pjrt_api",
"@xla//xla/pjrt:pjrt_c_api_client",
"@xla//xla/pjrt/cpu:cpu_client",
"@xla//xla:xla_proto_cc",
"@xla//xla:xla_proto_cc_impl",
"@xla//xla:xla_data_proto_cc",
"@xla//xla:xla_data_proto_cc_impl",
"@xla//xla/service:metrics_proto_cc",
"@xla//xla/service:metrics_proto_cc_impl",
"@xla//xla/service/cpu:cpu_compiler",
"@xla//xla/stream_executor/tpu:tpu_on_demand_compiler",
"@xla//xla/stream_executor/tpu:tpu_executor",
"@xla//xla/stream_executor/tpu:tpu_transfer_manager",
"@xla//xla/service/cpu:cpu_transfer_manager",
"@xla//xla/pjrt/gpu:se_gpu_pjrt_client",
"@xla//xla/tsl/protobuf:protos_all_cc_impl",
"@xla//xla/tsl/framework:allocator_registry_impl",
"@xla//xla/pjrt:status_casters",
"@xla//xla/python/ifrt",
"@xla//xla/python/pjrt_ifrt",
"@xla//xla/python/ifrt/hlo:hlo_program",
"@xla//xla/ffi:call_frame",
"@com_google_protobuf//:protobuf",
"@tsl//tsl/profiler/lib:profiler_session_impl",
"@tsl//tsl/profiler/lib:profiler_factory_impl",
"@tsl//tsl/profiler/lib:profiler_controller",
"@tsl//tsl/profiler/lib:traceme",
"@xla//xla/tsl/profiler/rpc:profiler_server_impl",
"@xla//xla/tsl/profiler/rpc/client:capture_profile",
"@xla//xla/tsl/profiler/rpc/client:profiler_client",
"@xla//xla/tsl/profiler/backends/cpu:annotation_stack_impl",
"@xla//xla/tsl/profiler/backends/cpu:traceme_recorder_impl",
"@xla//xla/tsl/profiler/utils:time_utils_impl",
"@tsl//tsl/profiler/protobuf:profiler_service_monitor_result_proto_cc_impl",
"@tsl//tsl/profiler/protobuf:profiler_service_proto_cc_impl",
"@tsl//tsl/profiler/protobuf:profiler_analysis_proto_cc_impl",
"@tsl//tsl/profiler/protobuf:profiler_options_proto_cc_impl",
"@tsl//tsl/profiler/protobuf:profile_proto_cc_impl",
"@tsl//tsl/profiler/protobuf:xplane_proto_cc_impl",
"@tsl//tsl/profiler/protobuf:trace_events_proto_cc_impl",
"@xla//xla/backends/profiler/cpu:host_tracer",
"@xla//xla/backends/profiler/cpu:host_tracer_impl",
"@xla//xla/backends/profiler/cpu:metadata_collector",
"@xla//xla/backends/profiler/cpu:metadata_utils",
"@tsl//tsl/platform:env_impl",
"@xla//xla/stream_executor:stream_executor_impl",
"@xla//xla/mlir/utils:type_util",
"@stablehlo//:stablehlo_capi_objects",
"@stablehlo//:chlo_capi_objects",
"@com_google_absl//absl/hash",
"@com_google_absl//absl/log:initialize",
"@com_google_absl//absl/log:globals",
"@llvm-project//mlir:CAPIIRObjects",
"@llvm-project//mlir:CAPILLVMObjects",
# "@jax//jaxlib/mosaic:tpu_dialect_capi_objects",
# "@jax//jaxlib/triton:triton_dialect_capi_objects",
],
)