[ROCm] reduce batch size to fix CI error (#15714)

ROCm CI batch size test occasionally fail. Try reduce batch size to fix
it.

error log:
Non-zero status code returned while running FusedMatMul node.
Name:'MatMul_2914_Grad/FusedMatMul_0' Status Message: HIP error
hipErrorNotFound:named symbol not found
Non-zero status code returned while running Gemm node.
Name:'MatMul_2891_Grad/Gemm_5' Status Message: HIP error
hipErrorNotFound:named symbol not found
This commit is contained in:
PeixuanZuo 2023-05-16 13:10:02 +08:00 committed by GitHub
parent bc5018a4e1
commit e96f10d27b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -56,7 +56,7 @@ def main():
configs["MI100_32G"] = [
Config(True, 128, 192, 20, ""),
Config(True, 512, 26, 80, ""),
Config(False, 128, 108, 20, ""),
Config(False, 128, 106, 20, ""),
Config(False, 512, 16, 80, ""),
]