Signed-off-by: Connor Carpenter <connorc@nvidia.com> Co-authored-by: Nick Hill <nickhill123@gmail.com>
22 lines
697 B
Rust
22 lines
697 B
Rust
// SPDX-License-Identifier: Apache-2.0
|
|
// SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
fn main() -> Result<(), Box<dyn std::error::Error>> {
|
|
let manifest_dir = std::env::var("CARGO_MANIFEST_DIR").unwrap();
|
|
let proto_dir = format!("{manifest_dir}/../../proto");
|
|
|
|
tonic_prost_build::configure()
|
|
.build_server(true)
|
|
.build_client(true)
|
|
.protoc_arg("--experimental_allow_proto3_optional") // be compatible with old compilers
|
|
.compile_protos(
|
|
&[
|
|
format!("{proto_dir}/control.proto"),
|
|
format!("{proto_dir}/inference.proto"),
|
|
],
|
|
&[proto_dir],
|
|
)?;
|
|
|
|
Ok(())
|
|
}
|