fix fp32 invalid when amp used by default & support amp arg

test_tipc/common_func.sh

@@ -29,9 +29,11 @@ function func_set_amp_params(){
     value=$2
 
     if [[ ${value} = "fp16" ]];then
-        echo "-o AMP.scale_loss=128 -o AMP.use_dynamic_loss_scaling=True -o AMP.level=O2"
+        echo "-o AMP.use_amp=True -o AMP.scale_loss=128 -o AMP.use_dynamic_loss_scaling=True -o AMP.level=O2"
+    elif [[ ${value} = "amp" ]];then
+        echo "-o AMP.use_amp=True -o AMP.scale_loss=128 -o AMP.use_dynamic_loss_scaling=True -o AMP.level=O1"
     else
-        echo " "
+        echo "-o AMP.use_amp=False"
     fi
 }