add gpu test in it

test/config/examples/mnist-pytorch-gpu.yml

@@ -0,0 +1,26 @@
+authorName: nni
+experimentName: default_test
+maxExecDuration: 15m
+maxTrialNum: 2
+trialConcurrency: 2
+searchSpacePath: ./mnist_pytorch_search_space.json
+
+tuner:
+  builtinTunerName: Random
+assessor:
+  builtinAssessorName: Medianstop
+  classArgs:
+    optimize_mode: maximize
+trial:
+  codeDir: ../../../examples/trials/mnist-pytorch
+  command: python3 mnist.py  --epochs 1 --batch_num 10
+  gpuNum: 1
+localConfig:
+  useActiveGpu: true
+  maxTrialNumPerGpu: 2
+
+useAnnotation: false
+multiPhase: false
+multiThread: false
+
+trainingServicePlatform: local

test/config/integration_tests.yml

@@ -76,6 +76,12 @@ testCases:
   launchCommand: python3 ../examples/trials/mnist-pytorch/mnist.py --epochs 1 --batch_num 0 --data_dir ../examples/trials/mnist-pytorch/data && nnictl create --config $configFile --debug
   trainingService: local
 
+- name: mnist-pytorch-local-gpu
+  configFile: test/config/examples/mnist-pytorch-gpu.yml
+  # download data first, to prevent concurrent issue.
+  launchCommand: python3 ../examples/trials/mnist-pytorch/mnist.py --epochs 1 --batch_num 0 --data_dir ../examples/trials/mnist-pytorch/data && nnictl create --config $configFile --debug
+  trainingService: local
+
 - name: mnist-pytorch-non-local
   configFile: test/config/examples/mnist-pytorch.yml
   # download data first, to prevent concurrent issue.