Sukumarsawant wrote: @tht2005 Let's skip these 3 functions for now. I don't think we should modify the inputs just for bf16 in a template that is also testing for other types. Rest LGTM. https://github.com/llvm/llvm-project/pull/218281