Use sizes rounded to 32

JuliaGPU · Nov 4, 2024 · ab3d38c · ab3d38c
1 parent 6a1820e
commit ab3d38c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tuning/wmma-contraction.jl b/tuning/wmma-contraction.jl
@@ -11,7 +11,7 @@ const accumulate_type = Float32
 
 const zero_c = true
 
-config_path = joinpath(@__DIR__, "../test/benchmark-suite.json")
+config_path = joinpath(@__DIR__, "../test/benchmark-suite-32.json")
 fp = open(config_path, "r")
 const jsonData = JSON.parse(read(fp, String))
 
@@ -192,7 +192,7 @@ function plot_best_configs(all_configs, best_configs)
     )
 
     p = plot()
-    title!("TTCG on $(name(device()))")
+    title!("TTCG on $(name(device())) (rounded to 32)")
     xlabel!("Tensor contraction")
     ylabel!("Performance relative to cuTENSOR [%]")