Add PyTorch data into performance comparison

InfiniTensor · Jan 14, 2025 · bec9334 · bec9334
1 parent 0dfaeb0
commit bec9334
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/performance_comparison.py b/performance_comparison.py
@@ -28,7 +28,7 @@ class KernelInformation:
     KernelInformation("attention", "attention-performance.csv", "Sequence Length"),
 )
 
-providers = ("Triton", "NineToothed")
+providers = ("Triton", "NineToothed", "PyTorch")
 
 num_rows = 2
 num_cols = 3