dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix bench load record benchmarks (#1826)

This commit is contained in:
Nathaniel Simard 2024-05-27 10:29:09 -04:00 committed by GitHub
parent cccd96de48
commit c59a3b8b8a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
backend-comparison/benches/load_record.rs
View File

@ -53,7 +53,7 @@ struct LoadRecordBenchmark<B: Backend> {
}
impl<B: Backend> Benchmark for LoadRecordBenchmark<B> {
type Args = BenchmarkModuleRecord<B>;
type Args = BenchmarkModule<B>;
fn name(&self) -> String {
format!("load_record_{:?}", self.kind).to_lowercase()
@ -67,7 +67,9 @@ impl<B: Backend> Benchmark for LoadRecordBenchmark<B> {
10
}
fn execute(&self, record: Self::Args) {
fn execute(&self, module: Self::Args) {
let record = module.into_record();
let _ = match self.kind {
Kind::Lazy => {
let module = self.config.init(&self.device);
@ -86,9 +88,8 @@ impl<B: Backend> Benchmark for LoadRecordBenchmark<B> {
fn prepare(&self) -> Self::Args {
let module = self.config.init(&self.device);
// Force sync.
let module_initialized = module.clone();
module_initialized.into_record()
module.clone()
}
fn sync(&self) {