@@ -11,12 +11,14 @@ fn main() {
1111 let dims: Dim4 = Dim4 :: new ( & [ 5 , 3 , 1 , 1 ] ) ;
1212
1313 println ! ( "Create a 5-by-3 matrix of random floats on the GPU" ) ;
14- let a: Array = af:: randu ( dims, af:: Aftype :: F32 ) ;
14+ let a: Array = af:: randu ( dims, af:: Aftype :: F32 ) . ok ( ) . unwrap ( ) ;
1515 af:: print ( & a) ;
1616
1717 println ! ( "Element-wise arithmetic" ) ;
18- let b: Array = & af:: sin ( & a) + 1.5 ;
19- let b2: Array = & af:: sin ( & a) + & af:: cos ( & a) ;
18+ let sin_res = af:: sin ( & a) . ok ( ) . unwrap ( ) ;
19+ let cos_res = af:: cos ( & a) . ok ( ) . unwrap ( ) ;
20+ let b: Array = & sin_res + 1.5 ;
21+ let b2: Array = & sin_res + & cos_res;
2022 let b3: Array = ! & a;
2123 println ! ( "sin(a) + 1.5 => " ) ; af:: print ( & b) ;
2224 println ! ( "sin(a) + cos(a) => " ) ; af:: print ( & b2) ;
@@ -30,7 +32,7 @@ fn main() {
3032 // af_print(B);
3133
3234 println ! ( "Fourier transform the result" ) ;
33- let c: Array = af:: fft ( & b, 1.0 , 0 ) ;
35+ let c: Array = af:: fft ( & b, 1.0 , 0 ) . ok ( ) . unwrap ( ) ;
3436 af:: print ( & c) ;
3537
3638 // printf("Grab last row\n");
@@ -49,12 +51,12 @@ fn main() {
4951
5052 // // Sort A
5153 println ! ( "Sort A and print sorted array and corresponding indices" ) ;
52- let ( vals, inds) = af:: sort_index ( & a, 0 , true ) ;
54+ let ( vals, inds) = af:: sort_index ( & a, 0 , true ) . ok ( ) . unwrap ( ) ;
5355 af:: print ( & vals) ;
5456 af:: print ( & inds) ;
5557
5658 println ! ( "u8 constant array" ) ;
57- let u8_cnst = af:: constant ( 1 as u8 , dims) ;
59+ let u8_cnst = af:: constant ( 1 as u8 , dims) . ok ( ) . unwrap ( ) ;
5860 af:: print ( & u8_cnst) ;
5961 println ! ( "Is u8_cnst array float precision type ? {}" , u8_cnst. is_single( ) . ok( ) . unwrap( ) ) ;
6062}
0 commit comments