Fix assertion in GQA backward example to ensure correct tensor comparison for gradient validation (#568)
Attach a file by drag & drop or click to upload
