Ok, but as the end goal is to use batched_gemm in the new back-end don't spend too more time on this op in the new gpu back-end