From daac74f25c6f32df3b75b371b25a7652059e6af9 Mon Sep 17 00:00:00 2001
From: x54-729 <17307130121@fudan.edu.cn>
Date: Wed, 18 May 2022 07:58:42 +0000
Subject: [PATCH] =?UTF-8?q?1.=E4=B8=BAtorch1.11=E7=9A=84=E6=B5=8B=E8=AF=95?=
 =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=85=B1=E4=BA=AB=E5=86=85=E5=AD=98=E4=B8=BA?=
 =?UTF-8?q?256M=202.prepare=5Fdataloader=E5=88=A4=E6=96=ADbackend=E6=97=B6?=
 =?UTF-8?q?=E6=B7=BB=E5=8A=A0dist-packages=E8=B7=AF=E5=BE=84?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Jenkinsfile                                    | 4 ++--
 fastNLP/core/dataloaders/prepare_dataloader.py | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/Jenkinsfile b/Jenkinsfile
index 20b48fbc..4ef51291 100644
--- a/Jenkinsfile
+++ b/Jenkinsfile
@@ -22,11 +22,11 @@ pipeline {
                     agent {
                         docker {
                             image 'fnlp:torch-1.11'
-                            args '-u root:root -v ${JENKINS_HOME}/html/docs:/docs -v ${JENKINS_HOME}/html/_ci:/ci --gpus all'
+                            args '-u root:root -v ${JENKINS_HOME}/html/docs:/docs -v ${JENKINS_HOME}/html/_ci:/ci --gpus all --shm-size 256M'
                         }
                     }
                     steps {
-                        sh 'pytest ./tests/ --durations=0 -m torch'
+                        sh 'pytest ./tests --durations=0 -m torch'
                     }
                 }
                 stage('Test Torch-1.6') {
diff --git a/fastNLP/core/dataloaders/prepare_dataloader.py b/fastNLP/core/dataloaders/prepare_dataloader.py
index 591c919c..81bd1bdb 100644
--- a/fastNLP/core/dataloaders/prepare_dataloader.py
+++ b/fastNLP/core/dataloaders/prepare_dataloader.py
@@ -91,6 +91,8 @@ def _check_module(module):
         for backend in SUPPORT_BACKENDS:
             if f'{os.sep}site-packages{os.sep}{backend}' in file:
                 return backend
+            if f'{os.sep}dist-packages{os.sep}{backend}' in file:
+                return backend
     except:
         pass
     return None