precision to args

undertherain · Jul 6, 2020 · 1e5c84f · 1e5c84f
1 parent ba918ea
commit 1e5c84f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/benchmarker/modules/do_cublas.py b/benchmarker/modules/do_cublas.py
@@ -6,6 +6,7 @@
 class Benchmark(IGEMM):
     def __init__(self, params, remaining_args=None):
         super().__init__(params, remaining_args)
+        assert self.params["problem"]["precision"] in ["FP32", "mixed"]
 
     def run(self):
         if "nb_gpus" in self.params: