MCPcopy Index your code

hub / github.com/NVlabs/Sana / endpoints

Endpoints11 in github.com/NVlabs/Sana

⨍ Functions 2,606◇ Types & classes 463↳ Endpoints 11
Routeattn_matmul
torch.amp.autocast("cuda", enabled=os.environ.get("AUTOCAST_LINEAR_ATTN", False) == "true")
diffusion/model/nets/sana_blocks.py:None
Routeattn_matmul
torch.cuda.amp.autocast(enabled=os.environ.get("AUTOCAST_LINEAR_ATTN", False) == "true")
diffusion/model/nets/fastlinear/modules/lite_mla.py:None
Routedepthwise_conv_fwd_kernel
triton.autotune( configs=get_autotune_config(), key=["B", "H", "W", "C", "K"], )
diffusion/model/nets/fastlinear/modules/triton_mb_conv_pre_glu_kernels/depthwise_conv_fwd.py:None
Routelinear_glu_fwd_kernel
custom_autotune( configs=get_autotune_config(), key=["M", "N", "K"], )
diffusion/model/nets/fastlinear/modules/triton_mb_conv_pre_glu_kernels/linear_glu_fwd.py:None
Routelinear_relu_fwd_kernel
custom_autotune( configs=get_autotune_config(), key=["M", "N", "K"], )
diffusion/model/nets/fastlinear/modules/triton_lite_mla_kernels/linear_relu_fwd.py:None
Routematmul_kernel
triton.autotune( configs=get_autotune_config(), key=["M", "N", "K"], )
diffusion/model/nets/fastlinear/modules/triton_lite_mla_kernels/mm.py:None
Routepad_vk_mm_fwd_kernel_fp32_fp32
custom_autotune( configs=get_autotune_config(), key=["B", "N", "H", "C"], )
diffusion/model/nets/fastlinear/modules/triton_lite_mla_kernels/pad_vk_mm_fwd.py:None
Routeproj_divide_bwd_kernel
custom_autotune( configs=get_autotune_config(), key=["M", "N", "K", "H_", "C_"], )
diffusion/model/nets/fastlinear/modules/triton_lite_mla_kernels/proj_divide_bwd.py:None
Routevk_mm_relu_bwd_kernel
custom_autotune( configs=get_autotune_config(), key=["B", "N", "H", "C"], )
diffusion/model/nets/fastlinear/modules/triton_lite_mla_kernels/vk_mm_relu_bwd.py:None
Routevk_q_mm_divide_fwd_kernel_fp32
custom_autotune( configs=get_autotune_config(), key=["B", "N", "H", "D"], )
diffusion/model/nets/fastlinear/modules/triton_lite_mla_kernels/vk_q_mm_divide_fwd.py:None
Routevk_q_mm_relu_bwd_kernel
custom_autotune( configs=get_autotune_config(), key=["B", "N", "H", "C"], )
diffusion/model/nets/fastlinear/modules/triton_lite_mla_kernels/vk_q_mm_relu_bwd.py:None