Merge pull request #28 from loliverhennigh/backend_exposure

xlb refactor
Autodesk · Dec 15, 2023 · 034ee36 · 034ee36
2 parents 2d6b799 + b3acb4a
commit 034ee36
Show file tree

Hide file tree

Showing 54 changed files with 4,734 additions and 1,813 deletions.
diff --git a/examples/backend_comparisons/README.md b/examples/backend_comparisons/README.md
@@ -0,0 +1,14 @@
+# Performance Comparisons
+
+This directory contains a minimal LBM implementation in Warp, Numba, and Jax. The
+code can be run with the following command:
+
+```bash
+python3 lattice_boltzmann.py
+```
+
+This will give MLUPs numbers for each implementation. The Warp implementation
+is the fastest, followed by Numba, and then Jax.
+
+This example should be used as a test for properly implementing more backends in
+XLB.