Endpoints — github.com/FlashSampling/FlashSampling

CodeHub MCPcopy Create free account

hub / github.com/FlashSampling/FlashSampling / endpoints

Endpoints9 in github.com/FlashSampling/FlashSampling

⨍ Functions 315◇ Types & classes 25 ↳ Endpoints 9

Routefused_mm_sample_triton_kernel

triton.autotune( configs=get_autotuning_configs(), key=["vocab_size", "hidden_size", "BLOCK_SIZE_H", "

src/fused_mm_sampling/core.py:None

Routefused_sample_helion_kernel

helion.kernel(autotune_effort=os.environ.get("HELION_AUTOTUNE_EFFORT", "quick"))

src/fused_mm_sampling/helion_impl.py:None

Routematmul_kernel

triton.autotune( configs=matmul_get_configs(), key=["M", "N", "K"], )

src/fused_mm_sampling/persistent_matmul.py:None

Routematmul_kernel_descriptor_persistent

triton.autotune( configs=matmul_tma_persistent_get_configs(), key=["M", "N", "K", "WARP_SPECIALIZE", "

src/fused_mm_sampling/persistent_matmul.py:None

Routematmul_kernel_persistent

triton.autotune( configs=matmul_get_configs(), key=["M", "N", "K"], )

src/fused_mm_sampling/persistent_matmul.py:None

Routematmul_kernel_tma

triton.autotune( configs=matmul_get_configs(pre_hook=matmul_tma_set_block_size_hook), key=["M", "N", "

src/fused_mm_sampling/persistent_matmul.py:None

Routematmul_kernel_tma_persistent

triton.autotune( configs=matmul_tma_persistent_get_configs(pre_hook=matmul_tma_set_block_size_hook), k

src/fused_mm_sampling/persistent_matmul.py:None

Routetest_greedy_tp2

pytest.mark.skipif( not os.environ.get("FMMS_TEST_DISTRIBUTED"), reason="Set FMMS_TEST_DISTRIBUTED=1 t

tests/test_core.py:None

Routetest_sampling_distribution_tp2

pytest.mark.skipif( not os.environ.get("FMMS_TEST_DISTRIBUTED"), reason="Set FMMS_TEST_DISTRIBUTED=1 t

tests/test_core.py:None