Browse Source

test(mge): do not ignore data test

GitOrigin-RevId: 937d24df5e
tags/v1.3.1
Megvii Engine Team 4 years ago
parent
commit
0ad85a41ac
3 changed files with 4 additions and 2 deletions
  1. +2
    -2
      imperative/python/megengine/data/dataloader.py
  2. +0
    -0
      imperative/python/megengine/data/tools/_queue.py
  3. +2
    -0
      imperative/python/test/run.sh

+ 2
- 2
imperative/python/megengine/data/dataloader.py View File

@@ -230,7 +230,7 @@ class _ParallelMapDataLoaderIter(_BaseMapDataLoaderIter):
]

# use shared-memory queue implemented by pyarrow plasma store.
from ._queue import PlasmaShmQueue
from .tools._queue import PlasmaShmQueue

self.batch_queue = PlasmaShmQueue(maxsize=2)

@@ -458,7 +458,7 @@ class _ParallelStreamDataLoaderIter(_BaseStreamDataLoaderIter):
]

# shared-memory queue implemented by pyarrow plasma store
from ._queue import PlasmaShmQueue
from .tools._queue import PlasmaShmQueue

self.batch_queue = PlasmaShmQueue(maxsize=2)



imperative/python/megengine/data/_queue.py → imperative/python/megengine/data/tools/_queue.py View File


+ 2
- 0
imperative/python/test/run.sh View File

@@ -13,6 +13,8 @@ else
exit 1
fi

export MEGENGINE_LOGGING_LEVEL="ERROR"

pushd $(dirname "${BASH_SOURCE[0]}")/.. >/dev/null
PYTHONPATH="." PY_IGNORE_IMPORTMISMATCH=1 python3 -m pytest $test_dirs -m 'not isolated_distributed'
if [[ "$TEST_PLAT" == cuda ]]; then


Loading…
Cancel
Save