You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+13-1Lines changed: 13 additions & 1 deletion
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -1,3 +1,15 @@
1
+
## BackendBench
2
+
3
+
A lot of people are now interested in optimizing existing kernels in PyTorch. This audience includes both systems researchers experimenting with new DSLs and LLM researchers looking to automate kernel authoring completely. But many existing efforts have been plagued by how to ensure correctness.
4
+
5
+
Our take is that if a kernel can replace an existing PyTorch operator and be merged into PyTorch's official codebase then it's far more likely to be correct but hacking on PyTorch's kernels has historically been challenging.
6
+
7
+
BackendBench is an evaluation suite that tests how good LLMs and humans are at writing a full fledged PyTorch backend. We make it possible for developers to add their custom kernels in well organized directory structure and dynamically override the core PyTorch aten operators at runtime. The outcome is a fully functional readable PyTorch backend you can pip install and run real models on with no model changes!
8
+
9
+
We provide both
10
+
1. Comprehensive operator level correctness checks using the PyTorch OpInfo test suite
11
+
2. Performance checks using the ops that show up in the most popular Hugging Face models with realistic tensor shapes
12
+
1
13
# Usage:
2
14
3
15
Run a simple smoke test (relu) with the default ATen backend:
@@ -33,4 +45,4 @@ Run LLM evaluation on smoke test (relu operation):
0 commit comments