alt extend

Author: VitaliiH

crates/prover/src/core/backend/icicle/circle.rs

@@ -103,11 +103,20 @@ impl PolyOps for IcicleBackend {
         folded
     }
 
+    // fn extend(poly: &CirclePoly<Self>, log_size: u32) -> CirclePoly<Self> {
+    //     assert!(log_size >= poly.log_size());
+    //     let count_zeros_to_extend = poly.coeffs.len();
+    //     let coeffs = DeviceVec::cuda_malloc_extend_with_zeros(&poly.coeffs.data, count_zeros_to_extend).unwrap();
+    //     CirclePoly::new(DeviceColumn {data: coeffs})
+    // }
+
     fn extend(poly: &CirclePoly<Self>, log_size: u32) -> CirclePoly<Self> {
         assert!(log_size >= poly.log_size());
-        let count_zeros_to_extend = poly.coeffs.len();
-        let coeffs = DeviceVec::cuda_malloc_extend_with_zeros(&poly.coeffs.data, count_zeros_to_extend).unwrap();
-        CirclePoly::new(DeviceColumn {data: coeffs})
+
+        let mut device_column = DeviceColumn::zeros(1 << log_size);
+        poly.coeffs.data.copy_to_device(&mut device_column.data).unwrap();
+
+        CirclePoly::new(device_column)
     }
 
     fn evaluate(