fix

atomicarchitects · Aug 14, 2024 · 13eccf2 · 13eccf2
1 parent 0e95db7
commit 13eccf2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/Makefile b/Makefile
@@ -12,7 +12,7 @@ profile:
 	nsys profile --capture-range=cudaProfilerApi --cuda-graph-trace=node --capture-range-end=stop -o profile -f true python train.py
 
 run:
-	./build/inference export/model.so
+	./build/inference ${PWD}/export/model.so
 
 .PHONY: run nsys build clean
 
diff --git a/README.md b/README.md
@@ -1,8 +1,7 @@
 ## TODO
 
-- [x] Get `torch.export` pipeline working.
-- [ ] Train tetris
-- [ ] Port `allegro-jax`
-
+- [x] Train tetris
+- [x] Get `torch.export` pipeline working
+- [] Call mode.so successfully in cpp