search

google / aqt

Apache License 2.0
247 stars 25 forks source link

Make local aqt support tiling the largest contraction axis (not the first) in the bwd. #663

Closed copybara-service[bot] closed 2 months ago

copybara-service[bot] commented 2 months ago

Make local aqt support tiling the largest contraction axis (not the first) in the bwd.