@triton.heuristics({"HAS_DT_BIAS": lambda args: args["dt_bias_ptr"] is not None}) @triton.heuristics({"HAS_D": lambda args: args["D_ptr"] is not None}) @triton ...
# Copyright (c) Huawei Technologies Co., Ltd. 2025. All rights reserved. This is a Triton implementation of the Flash Attention v2 algorithm from Tri Dao (https ...
Some results have been hidden because they may be inaccessible to you
Show inaccessible results