arnaucube/poulpy
1
0
Fork 0
mirror of https://github.com/arnaucube/poulpy.git synced 2026-02-10 13:16:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

Added memory layout field

Browse Source
This commit is contained in:
Jean-Philippe Bossuat
2025-04-25 08:57:08 +02:00
parent f64d786819
commit f0eaddb63e
3 changed files with 17 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
base2k/src/vec_znx.rs
View File

@@ -1,6 +1,7 @@
use crate::cast_mut;
use crate::ffi::vec_znx;
use crate::ffi::znx;
use crate::LAYOUT;
use crate::{Infos, Module};
use crate::{alloc_aligned, assert_alignement};
use itertools::izip;
@@ -24,6 +25,9 @@ pub struct VecZnx {
/// Stack size
pub size: usize,
/// Stacking layout
pub layout: LAYOUT,
/// Number of columns.
pub cols: usize,
@@ -59,6 +63,7 @@ impl VecZnx {
n: n,
size: size,
cols: cols,
layout: LAYOUT::COL,
data: Vec::from_raw_parts(ptr, bytes.len(), bytes.len()),
ptr: ptr,
}
@@ -76,6 +81,7 @@ impl VecZnx {
n: n,
size: size,
cols: cols,
layout: LAYOUT::COL,
data: Vec::new(),
ptr: bytes.as_mut_ptr() as *mut i64,
}
@@ -254,6 +260,7 @@ impl VecZnx {
Self {
n: n,
size: size,
layout: LAYOUT::COL,
cols: cols,
data: data,
ptr: ptr,