Skip to content

[TPU][Pallas] relax tolerances and fix Pallas autotuning OOM in layer_norm#2272

Draft
yarongmu-google wants to merge 1 commit into
pytorch:mainfrom
yarongmu-google:fix_layer_norm_tol
Draft

[TPU][Pallas] relax tolerances and fix Pallas autotuning OOM in layer_norm#2272
yarongmu-google wants to merge 1 commit into
pytorch:mainfrom
yarongmu-google:fix_layer_norm_tol