Ctrl+K

1 contributor

History: 1 commit

blake-snc

Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use

02fc236 verified 3 months ago

__pycache__
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
AUTHORS

29 Bytes
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
LICENSE

1.56 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
__init__.py

955 Bytes
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
ampere_helpers.py

3.71 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
barrier.py

2.05 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
bench_utils.py

8.35 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
benchmark.py

7.37 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
benchmark_flash_attention_fp8.py

16.7 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
blackwell_helpers.py

49.7 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
block_info.py

6.16 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
block_sparse_utils.py

58.3 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
block_sparsity.py

17.4 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
cache_utils.py

9.96 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
compute_block_sparsity.py

14.1 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
copy_utils.py

12.7 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
cute_dsl_ptxas.py

5.12 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
cute_dsl_utils.py

5.54 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
dropout.py

14.1 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
fa_logging.py

2.92 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
fast_math.py

492 Bytes
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
flash_bwd.py

68.7 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
flash_bwd_postprocess.py

27.2 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
flash_bwd_preprocess.py

15.5 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
flash_bwd_sm100.py

184 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
flash_bwd_sm120.py

2.03 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
flash_bwd_sm90.py

84.6 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
flash_fwd.py

79 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
flash_fwd_combine.py

31.4 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
flash_fwd_mla_sm100.py

144 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
flash_fwd_sm100.py

153 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
flash_fwd_sm120.py

3.41 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
flash_fwd_sm120_tma.py

41.7 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
flash_fwd_sm90.py

70.1 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
interface.py

96.5 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
mask.py

36.2 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
mma_sm100_desc.py

10.9 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
named_barrier.py

1.6 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
pack_gqa.py

11.6 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
paged_kv.py

9.09 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
pipeline.py

14.5 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
seqlen_info.py

11.1 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
sm90_config_search.py

14 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
softmax.py

27 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
testing.py

17.4 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
tile_scheduler.py

45.9 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
topk_gather_kv.py

10.2 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago
utils.py

30.6 kB
Restructure: build/torch-cuda/ layout with relative imports for self-contained kernels.get_kernel() use 3 months ago