Hello, can anybody point me to the source code for the triton kernel invoked by flashattention in torch.nn.functional.scaled_dot_product_attention? I cannot locate it. Thanks.
Hello, can anybody point me to the source code for the triton kernel invoked by flashattention in torch.nn.functional.scaled_dot_product_attention? I cannot locate it. Thanks.