github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-08 19:44:57 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat(rust): add from_tensor_[u32|u16]

Browse Source
This commit is contained in:
tmontaigu
2023-01-12 11:19:57 +01:00
parent c9f3983577
commit 6f0fbc8a3d
1 changed files with 28 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

28
compiler/lib/Bindings/Rust/src/compiler.rs
View File

@@ -781,6 +781,34 @@ impl LambdaArgument {
}
}
pub fn from_tensor_u16(data: &[u16], dims: &[i64]) -> Result<LambdaArgument, CompilerError> {
unsafe {
let arg = LambdaArgument::wrap(ffi::lambdaArgumentFromTensorU16(
data.as_ptr(),
dims.as_ptr(),
dims.len().try_into().unwrap(),
));
if arg.is_null() {
return Err(CompilerError(arg.error_msg()));
}
Ok(arg)
}
}
pub fn from_tensor_u32(data: &[u32], dims: &[i64]) -> Result<LambdaArgument, CompilerError> {
unsafe {
let arg = LambdaArgument::wrap(ffi::lambdaArgumentFromTensorU32(
data.as_ptr(),
dims.as_ptr(),
dims.len().try_into().unwrap(),
));
if arg.is_null() {
return Err(CompilerError(arg.error_msg()));
}
Ok(arg)
}
}
pub fn from_tensor_u64(data: &[u64], dims: &[i64]) -> Result<LambdaArgument, CompilerError> {
unsafe {
let arg = LambdaArgument::wrap(ffi::lambdaArgumentFromTensorU64(