chore(gpu): fix gpu package for publication

This commit is contained in:
Agnes Leroy
2024-01-22 13:52:02 +01:00
committed by Arthur Meyre
parent f313b58c8e
commit 548f2e5d05
80 changed files with 4 additions and 4 deletions

View File

@@ -144,7 +144,7 @@ fmt: install_rs_check_toolchain
.PHONY: fmt_gpu # Format rust and cuda code
fmt_gpu: install_rs_check_toolchain
cargo "$(CARGO_RS_CHECK_TOOLCHAIN)" fmt
cd backends/tfhe-cuda-backend/implementation/ && ./format_tfhe_cuda_backend.sh
cd "$(TFHECUDA_SRC)" && ./format_tfhe_cuda_backend.sh
.PHONY: check_fmt # Check rust code format
check_fmt: install_rs_check_toolchain

View File

@@ -1,6 +1,6 @@
BSD 3-Clause Clear License
Copyright © 2023 ZAMA.
Copyright © 2024 ZAMA.
All rights reserved.
Redistribution and use in source and binary forms, with or without modification,

View File

@@ -12,7 +12,7 @@ fn main() {
Only Ubuntu is supported by tfhe-cuda-backend at this time. Build may fail\n"
);
}
let dest = cmake::build("../implementation");
let dest = cmake::build("cuda");
println!("cargo:rustc-link-search=native={}", dest.display());
println!("cargo:rustc-link-lib=static=tfhe_cuda_backend");
println!("cargo:rustc-link-search=native=/usr/local/cuda/lib64");

View File

@@ -62,7 +62,7 @@ rayon = { version = "1.5.0" }
bincode = "1.3.3"
concrete-fft = { version = "0.3.0", features = ["serde", "fft128"] }
pulp = "0.13"
tfhe-cuda-backend = {path = "../backends/tfhe-cuda-backend/rust_api", optional = true}
tfhe-cuda-backend = {path = "../backends/tfhe-cuda-backend", optional = true}
aligned-vec = { version = "0.5", features = ["serde"] }
dyn-stack = { version = "0.9" }
paste = "1.0.7"