Skip to content

Commit 65cc4ff

Browse files
committed
Fix pad FFI arguments in wrapper code
1 parent bce4b85 commit 65cc4ff

File tree

1 file changed

+12
-2
lines changed

1 file changed

+12
-2
lines changed

src/core/data.rs

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -948,9 +948,9 @@ pub fn pad<T: HasAfEnum>(
948948
let err_val = af_pad(
949949
&mut temp as *mut af_array,
950950
input.get(),
951-
begin.ndims() as c_uint,
951+
4,
952952
begin.get().as_ptr() as *const dim_t,
953-
end.ndims() as c_uint,
953+
4,
954954
end.get().as_ptr() as *const dim_t,
955955
fill_type as c_uint,
956956
);
@@ -963,7 +963,9 @@ pub fn pad<T: HasAfEnum>(
963963
mod tests {
964964
use super::reorder_v2;
965965

966+
use super::super::defines::BorderType;
966967
use super::super::random::randu;
968+
use super::pad;
967969

968970
use crate::dim4;
969971

@@ -976,4 +978,12 @@ mod tests {
976978
let _swap_1_2 = reorder_v2(&a, 0, 2, Some(vec![1]));
977979
let _swap_0_3 = reorder_v2(&a, 3, 1, Some(vec![2, 0]));
978980
}
981+
982+
#[test]
983+
fn check_pad_api() {
984+
let a = randu::<f32>(dim4![3, 3]);
985+
let begin_dims = dim4!(0, 0, 0, 0);
986+
let end_dims = dim4!(2, 2, 0, 0);
987+
let _padded = pad(&a, begin_dims, end_dims, BorderType::ZERO);
988+
}
979989
}

0 commit comments

Comments
 (0)