update max_abs_err to account for other kernels not only loss kernels
update max_abs_err to account for other kernels not only loss kernels