[AMD] [ROCm] Pick num_warps
based on platform
#326
+524
−9
num_warps
based on platform
#326