fix ci error (#10295)

修复第三方库升级导致的CI问题，将CI中使用的tensorboradX版本设置为v2.6 将TestGlobalPoolHigherDerivative设置为1n1d，否则会出现 out of memory
Oneflow-Inc · Jun 26, 2023 · 1876b0d · 1876b0d
1 parent ae52678
commit 1876b0d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
@@ -583,6 +583,7 @@ jobs:
         run: |
           docker exec ${TEST_CONTAINER_NAME} python3 -m pip install -e ${{ env.FLOW_VISION_SRC}}
           docker exec ${TEST_CONTAINER_NAME} python3 -m pip install pybind11 --user
+          docker exec ${TEST_CONTAINER_NAME} python3 -m pip install tensorboardX==2.6 --user
           docker exec ${TEST_CONTAINER_NAME} python3 -m pip install -e ${{ env.LIBAI_SRC}}
           docker exec ${TEST_CONTAINER_NAME} python3 -m pip install -e ${{ env.ONEFLOW_IREE_SRC}}
           docker exec ${TEST_CONTAINER_NAME} python3 -m pip install -e ${{ env.ONE_FX_SRC}}
@@ -899,6 +900,7 @@ jobs:
         run: |
           docker exec ${TEST_CONTAINER_NAME} python3 -m pip install -e ${{ env.FLOW_VISION_SRC}}
           docker exec ${TEST_CONTAINER_NAME} python3 -m pip install pybind11 --user
+          docker exec ${TEST_CONTAINER_NAME} python3 -m pip install tensorboardX==2.6 --user
           docker exec ${TEST_CONTAINER_NAME} python3 -m pip install -e ${{ env.LIBAI_SRC}}
           docker exec ${TEST_CONTAINER_NAME} python3 -m pip install -e ${{ env.ONEFLOW_FACE_SRC}}
           docker exec ${TEST_CONTAINER_NAME} python3 -m pip install -e ${{ env.ONEFLOW_IREE_SRC}}

diff --git a/python/oneflow/test/modules/test_global_higher_derivative_pool.py b/python/oneflow/test/modules/test_global_higher_derivative_pool.py
@@ -117,6 +117,7 @@ def _test_adaptive_pool_grad_grad_impl(test_case, placement, ndim, mode):
     _check_equal(test_case, ddy.pytorch, ddy.oneflow, "ddy")
 
 
+@flow.unittest.skip_unless_1n1d()
 class TestGlobalPoolHigherDerivative(flow.unittest.TestCase):
     @globaltest
     def test_max_pool_1d_grad_grad(test_case):