Update headers

jbrhm · Jul 17, 2024 · 8a6fb17 · 8a6fb17
1 parent 785c44b
commit 8a6fb17
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/testing/pytorch.py b/testing/pytorch.py
@@ -16,7 +16,7 @@ class Backends(Enum):
     cupybara = 1
     torch = 2
 
-backend = Backends.torch
+backend = Backends.cupybara
 
 begin = time.time()
 
@@ -46,4 +46,4 @@ class Backends(Enum):
 
 GFLOPS = ((2 * SIZE * SIZE * SIZE) / (end - begin)) * 1e-9; 
 
-print(f"Pytorch had {GFLOPS} GFLOPS")
+print(f"{GFLOPS} GFLOPS")