fixed rocm batch_matmul strategy for mixed i8i8i32

Author: masahi

python/tvm/relay/op/strategy/rocm.py

@@ -212,7 +212,7 @@ def batch_matmul_strategy_rocm(attrs, inputs, out_type, target):
     """Batch matmul strategy for ROCM"""
     strategy = _op.OpStrategy()
     strategy.add_implementation(
-        wrap_compute_batch_matmul(topi.cuda.batch_matmul),
+        wrap_compute_batch_matmul(topi.cuda.batch_matmul, need_out_dtype=True),
         wrap_topi_schedule(topi.cuda.schedule_batch_matmul),
         name="batch_matmul.cuda",
         plevel=10,