Skip to content

Allocate dQ, dK, and dV as a catted tensor to save a downstream cat in nvFuser. #39

Allocate dQ, dK, and dV as a catted tensor to save a downstream cat in nvFuser.

Allocate dQ, dK, and dV as a catted tensor to save a downstream cat in nvFuser. #39

Annotations

1 warning

auto-cc

succeeded Mar 26, 2024 in 14s