Skip to content

Commit 51d7ed8

Browse files
committed
unsafe
Signed-off-by: Nicholas Gates <[email protected]>
1 parent 79b5719 commit 51d7ed8

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

vortex-compute/src/take/slice/avx2.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -72,7 +72,7 @@ pub unsafe fn take_avx2<V: NativePType, I: UnsignedPType>(
7272
let values = unsafe { std::mem::transmute::<&[V], &[$cast]>(buffer) };
7373

7474
let result = exec_take::<$cast, $indices, AVX2Gather>(values, indices);
75-
result.transmute::<V>()
75+
unsafe { result.transmute::<V>() }
7676
}};
7777
}
7878

vortex-compute/src/take/slice/portable.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,21 +5,21 @@
55
66
#![cfg(vortex_nightly)]
77

8-
use std::mem::MaybeUninit;
98
use std::mem::size_of;
109
use std::mem::transmute;
10+
use std::mem::MaybeUninit;
1111
use std::simd;
1212
use std::simd::num::SimdUint;
1313

1414
use multiversion::multiversion;
1515
use vortex_buffer::Alignment;
1616
use vortex_buffer::Buffer;
1717
use vortex_buffer::BufferMut;
18+
use vortex_dtype::match_each_native_simd_ptype;
19+
use vortex_dtype::match_each_unsigned_integer_ptype;
1820
use vortex_dtype::NativePType;
1921
use vortex_dtype::PType;
2022
use vortex_dtype::UnsignedPType;
21-
use vortex_dtype::match_each_native_simd_ptype;
22-
use vortex_dtype::match_each_unsigned_integer_ptype;
2323

2424
/// SIMD types larger than the SIMD register size are beneficial for
2525
/// performance as this leads to better instruction level parallelism.
@@ -40,7 +40,7 @@ pub fn take_portable<T: NativePType, I: UnsignedPType>(buffer: &[T], indices: &[
4040
// make.
4141
let u16_slice: &[u16] =
4242
unsafe { std::slice::from_raw_parts(buffer.as_ptr() as *const u16, buffer.len()) };
43-
return take_with_indices(u16_slice, indices).transmute::<T>();
43+
return unsafe { take_with_indices(u16_slice, indices).transmute::<T>() };
4444
}
4545

4646
match_each_native_simd_ptype!(T::PTYPE, |TC| {

0 commit comments

Comments
 (0)