Add PyTorch data into performance comparison

voltjia · voltjia · commit bec933468633 · 2025-01-14T14:31:00.000+08:00
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

Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,7 @@ class KernelInformation:`
`28`	`28`	`KernelInformation("attention", "attention-performance.csv", "Sequence Length"),`
`29`	`29`	`)`
`30`	`30`
`31`		`-providers = ("Triton", "NineToothed")`
	`31`	`+providers = ("Triton", "NineToothed", "PyTorch")`
`32`	`32`
`33`	`33`	`num_rows = 2`
`34`	`34`	`num_cols = 3`