diff --git a/xCAT-test/autotest/testcase/infiniband/cases0 b/xCAT-test/autotest/testcase/infiniband/cases0
index 65fbcc04e..c66b66ca7 100644
--- a/xCAT-test/autotest/testcase/infiniband/cases0
+++ b/xCAT-test/autotest/testcase/infiniband/cases0
@@ -1,3 +1,13 @@
+start:rhel_infiniband_diskless_install
+cmd:RHEL_ISO="$$ISO" LINUX_DISTRO="$$OS" COMPUTE_NODE="$$CN" MLNX_ISO="$$MLNX" DKMS_RPM="$$DKMS" /opt/xcat/share/xcat/tools/autotest/testcase/infiniband/rhel-infiniband-diskless-install.sh
+check:rc==0
+end
+
+start:rhel_infiniband_diskfull_install
+cmd:RHEL_ISO="$$ISO" LINUX_DISTRO="$$OS" COMPUTE_NODE="$$CN" MLNX_ISO="$$MLNX" DKMS_RPM="$$DKMS" /opt/xcat/share/xcat/tools/autotest/testcase/infiniband/rhel-infiniband-diskfull-install.sh
+check:rc==0
+end
+
 start:confignetwork_ib_ipoib
 description: this case is to test confignetwork could configure the first infiniband interface, ib0, successfully with updatenode.
 cmd:lsdef $$CN;if [ $? -eq 0 ]; then lsdef -l $$CN -z >/tmp/CN.standa ;fi
diff --git a/xCAT-test/autotest/testcase/infiniband/rhel-infiniband-diskfull-install.sh b/xCAT-test/autotest/testcase/infiniband/rhel-infiniband-diskfull-install.sh
index 6cc575646..66ceb74fe 100755
--- a/xCAT-test/autotest/testcase/infiniband/rhel-infiniband-diskfull-install.sh
+++ b/xCAT-test/autotest/testcase/infiniband/rhel-infiniband-diskfull-install.sh
@@ -3,9 +3,12 @@
 ########
 # Set all the variables below
 
+[ -n "$LINUX_DISTRO" ] &&
 LINUX_DISTRO="rhels7.4"
+[ -n "$LINUX_ARCH" ] &&
 LINUX_ARCH="ppc64le"
 
+[ -n "$COMPUTE_NODE" ] &&
 COMPUTE_NODE="c910f03c01p10"
 SOURCE_DIR="/media/xcat"
 
diff --git a/xCAT-test/autotest/testcase/infiniband/rhel-infiniband-diskless-install.sh b/xCAT-test/autotest/testcase/infiniband/rhel-infiniband-diskless-install.sh
index 18f61a70e..a5aad0e07 100755
--- a/xCAT-test/autotest/testcase/infiniband/rhel-infiniband-diskless-install.sh
+++ b/xCAT-test/autotest/testcase/infiniband/rhel-infiniband-diskless-install.sh
@@ -3,9 +3,12 @@
 ########
 # Set all the variables below
 
+[ -n "$LINUX_DISTRO" ] &&
 LINUX_DISTRO="rhels7.4"
+[ -n "$LINUX_ARCH" ] &&
 LINUX_ARCH="ppc64le"
 
+[ -n "$COMPUTE_NODE" ] &&
 COMPUTE_NODE="c910f03c01p10"
 SOURCE_DIR="/media/xcat"