Fix benchmark problem (#3762)

* fix

* fix bench

* fix

* Update bench_rgcn.py

* try fix

* fix wrong fix

* Update bench_rgcn.py

Jenkinsfile

@@ -107,9 +107,10 @@ pipeline {
   stages {
     stage('Regression Test Trigger') {
       agent {
-        kubernetes {
-          yamlFile 'docker/pods/ci-lint.yaml'
-          defaultContainer 'dgl-ci-lint'
+        docker {
+            label 'linux-benchmark-node'
+            image 'dgllib/dgl-ci-lint'
+            alwaysPull true
         }
       }
       when { triggeredBy 'IssueCommentCause' }

benchmarks/benchmarks/model_speed/bench_rgcn.py

@@ -41,7 +41,11 @@ class RGCN(nn.Module):
 def track_time(data, use_type_count):
     # args
     if data == 'aifb':
+        if dgl.__version__.startswith("0.8"):
+            num_bases = None
+        else:
             num_bases = -1
+            
         l2norm = 0.
     elif data == 'am':
         num_bases = 40

benchmarks/benchmarks/utils.py

@@ -251,7 +251,7 @@ class PinsageDataset:
 
 
 def load_nowplaying_rs():
-    import torchtext
+    import torchtext.legacy as torchtext
     # follow examples/pytorch/pinsage/README to create nowplaying_rs.pkl
     name = 'nowplaying_rs.pkl'
     dataset_dir = os.path.join(os.getcwd(), 'dataset')

benchmarks/scripts/publish.sh

@@ -55,14 +55,14 @@ if [[ $DEVICE == "cpu" ]]; then
         --rm \
         $DOCKER_MOUNT_OPT \
         $DOCKER_ENV_OPT \
-        --shm-size="4g" \
+        --shm-size="16g" \
         --hostname=$MACHINE -dit public.ecr.aws/s1o7b3d9/benchmakrk_pyg_dgl:cu111_torch181_pyg170 /bin/bash
 else
     docker run --name dgl-reg \
         --rm --gpus all \
         $DOCKER_MOUNT_OPT \
         $DOCKER_ENV_OPT \
-        --shm-size="4g" \
+        --shm-size="16g" \
         --hostname=$MACHINE -dit public.ecr.aws/s1o7b3d9/benchmakrk_pyg_dgl:cu111_torch181_pyg170 /bin/bash
 fi