@@ -7,8 +7,8 @@ use openvm_circuit::{
77} ;
88#[ cfg( feature = "cuda" ) ]
99use openvm_cuda_backend:: {
10- data_transporter:: transport_air_proving_ctx_to_device, BabyBearBn254Poseidon2GpuEngine ,
11- BabyBearBn254Poseidon2HashScheme ,
10+ data_transporter:: transport_air_proving_ctx_to_device, BabyBearBn254Poseidon2HashScheme ,
11+ GenericGpuBackend ,
1212} ;
1313use openvm_stark_backend:: {
1414 proof:: Proof ,
@@ -113,7 +113,7 @@ impl<SC: StarkProtocolConfig<F = F>> RootTraceGen<CpuBackend<SC>> for RootTraceG
113113}
114114
115115#[ cfg( feature = "cuda" ) ]
116- impl RootTraceGen < BabyBearBn254Poseidon2GpuEngine > for RootTraceGenImpl {
116+ impl RootTraceGen < GenericGpuBackend < BabyBearBn254Poseidon2HashScheme > > for RootTraceGenImpl {
117117 fn new ( deferral_enabled : bool ) -> Self {
118118 Self { deferral_enabled }
119119 }
@@ -124,7 +124,7 @@ impl RootTraceGen<BabyBearBn254Poseidon2GpuEngine> for RootTraceGenImpl {
124124 user_pvs_proof : & UserPublicValuesProof < DIGEST_SIZE , F > ,
125125 memory_dimensions : MemoryDimensions ,
126126 ) -> (
127- Vec < AirProvingContext < BabyBearBn254Poseidon2GpuEngine > > ,
127+ Vec < AirProvingContext < GenericGpuBackend < BabyBearBn254Poseidon2HashScheme > > > ,
128128 Vec < [ F ; POSEIDON2_WIDTH ] > ,
129129 ) {
130130 let ( cpu_ctxs, inputs) =
@@ -142,7 +142,7 @@ impl RootTraceGen<BabyBearBn254Poseidon2GpuEngine> for RootTraceGenImpl {
142142 memory_dimensions : MemoryDimensions ,
143143 deferral_merkle_proofs : Option < & DeferralMerkleProofs < F > > ,
144144 ) -> (
145- Vec < AirProvingContext < BabyBearBn254Poseidon2GpuEngine > > ,
145+ Vec < AirProvingContext < GenericGpuBackend < BabyBearBn254Poseidon2HashScheme > > > ,
146146 Vec < [ F ; POSEIDON2_WIDTH ] > ,
147147 ) {
148148 let ( cpu_ctxs, inputs) =
0 commit comments