Merge branch 'master' into luqi/autotest_on_complex

Oneflow-Inc · Jun 26, 2023 · 756bdef · 756bdef
2 parents 79231c1 + 1876b0d
commit 756bdef
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):