Merge pull request #64 from Bluefog-Lib/optimizer_test

Optimizer test
Bluefog-Lib · Dec 29, 2020 · 1779321 · 1779321
2 parents e8380bb + b172a7c
commit 1779321
Show file tree

Hide file tree

Showing 2 changed files with 536 additions and 1 deletion.
diff --git a/Makefile b/Makefile
@@ -15,7 +15,7 @@ build:
 	python setup.py build_ext -i
 
 test: test_torch
-test_torch: test_torch_basic test_torch_ops test_torch_win_ops
+test_torch: test_torch_basic test_torch_ops test_torch_win_ops test_torch_optimizer
 test_tensorflow: test_tensorflow_basic test_tensorflow_ops
 test_all: test_torch test_tensorflow
 
@@ -33,6 +33,10 @@ test_torch_ops:
 test_timeline:
 	${MPIRUN} ${PYTEST} ./test/timeline_test.py
 
+.PHONY: test_torch_optimizer
+test_torch_optimizer:
+	${MPIRUN} ${PYTEST} ./test/torch_optimizer_test.py
+
 .PHONY: test_torch_win_ops
 test_torch_win_ops:
 	${MPIRUN} ${PYTEST} ./test/torch_win_ops_test.py