Add tune_with marker macro

tracel-ai · Sep 19, 2024 · c1f9169 · c1f9169
1 parent 41ca9c0
commit c1f9169
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/crates/cubecl-runtime/src/tune/mod.rs b/crates/cubecl-runtime/src/tune/mod.rs
@@ -5,6 +5,7 @@ mod tune_cache;
 mod tuner;
 mod util;
 
+pub use crate::tune_with;
 pub use local::*;
 pub use operation::*;
 pub use tune_benchmark::*;

diff --git a/crates/cubecl-runtime/src/tune/util.rs b/crates/cubecl-runtime/src/tune/util.rs
@@ -12,3 +12,14 @@ pub fn anchor(x: usize, max: Option<usize>) -> usize {
         power_of_2
     }
 }
+
+/// Tune the operation set with these benchmark inputs
+#[macro_export]
+macro_rules! tune_with {
+    ($($args:expr),*) => {
+        ($($args),*)
+    };
+    ($($args:expr,)*) => {
+        ($($args),*)
+    };
+}