Update burnbenchapp

tracel-ai · Mar 27, 2024 · e9f32da · e9f32da
1 parent da186fd
commit e9f32da
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/backend-comparison/Cargo.toml b/backend-comparison/Cargo.toml
@@ -71,8 +71,9 @@ name = "data"
 harness = false
 
 [[bench]]
-name = "load_record"
+name = "load-record"
 harness = false
+path = "benches/load_record.rs"
 
 [[bench]]
 name = "custom-gelu"

diff --git a/backend-comparison/src/burnbenchapp/base.rs b/backend-comparison/src/burnbenchapp/base.rs
@@ -102,6 +102,8 @@ pub(crate) enum BenchmarkValues {
     Unary,
     #[strum(to_string = "max-pool2d")]
     MaxPool2d,
+    #[strum(to_string = "load-record")]
+    LoadRecord,
 }
 
 pub fn execute() {