Fix benchmark collection gradient check (#2564)

Fix benchmark collection gradient check (#2564) #7225