-
Notifications
You must be signed in to change notification settings - Fork 376
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
feat: integrate gpu into sdk #1301
Conversation
sdk/src/provers/local.rs
Outdated
return Err(anyhow::anyhow!( | ||
"not enough memory to generate plonk proof. at least 128GB is required." | ||
"not enough memory to generate plonk proof. at least 64GB is required." |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Should we mkae 60=64?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
made the threshold PLONK_MEMORY_GB_REQUIREMENT - 4, don't want to use 64gb exactly because lots of systems don't have 64GB exactly
sdk/src/provers/local.rs
Outdated
@@ -3,6 +3,9 @@ use sp1_core::runtime::SP1Context; | |||
use sp1_prover::{components::SP1ProverComponents, SP1Prover, SP1Stdin}; | |||
use sysinfo::System; | |||
|
|||
#[cfg(feature = "cuda")] | |||
use sp1_cuda::SP1CudaProver; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
nit: why don't you make a CudaProver
instead of LocalProver
and having the config flag stuff? that seems a bit cleaner
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
agreed that's cleaner, done
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
there is some repeated code but maybe its fine
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
i think eventually we should make SP1CudaProver
and SP1Prover
conform to a single unifying trait
No description provided.