ReduceMax wrong number of arguments for bert

[umangyadav]

DOR (Definition of Ready)

ready

Description

This is happening develop branch of MIGraphX.

This is the commit SHA

commit 2840377d81adab2ab11fb762907ac6f8ba214056 (HEAD -> develop, origin/develop, origin/HEAD)
Author: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Date:   Sun Aug 3 15:54:19 2025 -0400

    Update rocMLIR main ff95d9286f1c273c1adc7a04db248a018be15550 (#4198)

Steps to reproduce

1. Download Bert model : wget https://github.com/stefankoncarevic/Onnx_models/raw/master/bert_base_cased_1.onnx

2. Enable all MLIR fusions by setting these flags
   MIGRAPHX_MLIR_USE_SPECIFICS_OPS="fused,convolution,dot,attention"
   MIGRAPHX_ENABLE_MLIR_INPUT_FUSION=1
   MIGRAPHX_ENABLE_MLIR_REDUCE_FUSION=1
   MIGRAPHX_MLIR_ENABLE_SPLITK=1

3. Install accuracy checker requirements
   pip3 install -r tools/accuracy/requirements.txt

4. Run model
   python3 accuracy_checker.py --fill1 --onnx ../../bert_base_cased_1.onnx

It shows error :

Pass: gpu::compile_ops
terminate called after throwing an instance of 'migraphx::version_1::exception'
  what():  ../../../src/include/migraphx/check_shapes.hpp:119: has: reduce_max: Wrong number of arguments: expected 2 but given 1
terminate called recursively
terminate called recursively
terminate called recursively
terminate called recursively
terminate called recursively
terminate called recursively
terminate called recursively
terminate called recursively
terminate called recursively

Here is the log with MIGRAPHX_TRACE_COMPILE=1

DOD (Definition of Done)

Issue is fixed.

[umangyadav]

compile_logs.txt