ruPRIM-host 0.1.22

ruPRIM CPU tensor primitives, reductions, indexing and SIMD kernels.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15

    use super::*;
    use ruda_core::tensor::data::TensorData;

    #[test]
    fn test_flip_is_zero_copy() {
        // Verify flip doesn't copy data by checking it shares the same underlying storage
        let tensor = HostTensor::from_data(TensorData::new(vec![1.0f32, 2.0, 3.0, 4.0], [4]));
        let tensor_ptr = tensor.bytes().as_ptr();
        let flipped = flip(tensor, &[0]);
        let flipped_ptr = flipped.bytes().as_ptr();
        assert_eq!(
            tensor_ptr, flipped_ptr,
            "flip should share underlying storage"
        );
    }