MCPcopy Create free account

hub / github.com/AlphaGPU/leetgpu-challenges / functions

Functions919 in github.com/AlphaGPU/leetgpu-challenges

⨍ Functions 919◇ Types & classes 95
↓ 12 callersMethod_make_test_case
(self, seq_len, zero_x=False)
challenges/hard/74_gpt2_block/challenge.py:196
↓ 12 callersMethod_make_test_case
(self, seq_len, zero_x=False)
challenges/hard/93_llama_transformer_block/challenge.py:242
↓ 11 callersMethod_make_test_case
(self, batch, seq_len, d_model, d_state, zero_u=False, zero_delta=False)
challenges/medium/94_ssm_selective_scan/challenge.py:117
↓ 11 callersMethod_make_test_case
(self, M: int, N: int, K: int, group_size: int, zero_x: bool = False)
challenges/medium/81_int4_matmul/challenge.py:86
↓ 11 callersMethod_make_test_case
(self, num_heads, seq_len, head_dim, zero_q=False, seed=None)
challenges/medium/96_int8_kv_cache_attention/challenge.py:88
↓ 10 callersMethod_make_test_case
(self, num_q_heads, num_kv_heads, seq_len, head_dim, zero_inputs=False)
challenges/medium/80_grouped_query_attention/challenge.py:80
↓ 10 callersMethod_make_test_case
(self, B, L, zero_inputs=False, zero_a=False, unit_a=False)
challenges/medium/82_linear_recurrence/challenge.py:64
↓ 10 callersMethod_make_test_case
( self, seq_len: int, d_model: int, gamma: float = 0.9, zero_qk: bool
challenges/medium/92_decaying_causal_attention/challenge.py:79
↓ 10 callersMethod_make_test_case
(self, M, d_model, d_ffn, zero_x=False)
challenges/medium/84_swiglu_mlp_block/challenge.py:61
↓ 9 callersMethod_make_test
(self, M: int, N: int, lo: float = -10.0, hi: float = 10.0)
challenges/medium/71_parallel_merge/challenge.py:50
↓ 9 callersMethod_make_test_case
(self, batch, d_in, d_out, rank, lora_scale=0.5, zero_x=False)
challenges/medium/85_lora_linear/challenge.py:69
↓ 6 callersMethod_build_case
( self, token_ids: torch.Tensor, position_ids: torch.Tensor, token_embeddings:
challenges/medium/106_token_embedding_layer/challenge.py:73
↓ 6 callersMethod_random_case
( self, B: int, T: int, V: int, P: int, D: int, eps: f
challenges/medium/106_token_embedding_layer/challenge.py:144
↓ 5 callersMethod_make_test
(self, N: int, lo: float = -2.0, hi: float = 2.0)
challenges/medium/72_stream_compaction/challenge.py:40
↓ 5 callersMethod_make_test_case
(self, B, T, V, seed=42)
challenges/medium/87_speculative_decoding_verification/challenge.py:156
↓ 4 callersFunction_make_graph
Create a random non-negative weighted directed graph as a flat float32 tensor.
challenges/hard/73_all_pairs_shortest_paths/challenge.py:8
↓ 4 callersFunction_unit_rms_norm
(x: torch.Tensor)
challenges/medium/76_adder_transformer/challenge.py:80
↓ 3 callersFunction_init_weights
(device)
challenges/medium/76_adder_transformer/challenge.py:65
↓ 3 callersMethod_make_accept_all_case
All draft tokens accepted: target_probs == draft_probs so alpha == 1 everywhere.
challenges/medium/87_speculative_decoding_verification/challenge.py:187
↓ 3 callersMethod_make_reject_first_case
First draft token always rejected: draft_probs high, target low for that token.
challenges/medium/87_speculative_decoding_verification/challenge.py:218
↓ 3 callersMethod_make_sparse_probs
Generate sparse probability distributions: only K tokens have nonzero probability. Using sparse distributions ensures that the adjusted distr
challenges/medium/87_speculative_decoding_verification/challenge.py:136
↓ 2 callersFunction_encode_pair
(a: int, b: int)
challenges/medium/76_adder_transformer/challenge.py:45
↓ 2 callersMethodlayer_norm
(z, w, b, eps=1e-5)
challenges/hard/74_gpt2_block/challenge.py:103
↓ 1 callersFunction_encode_pairs_batch
(a_vals: torch.Tensor, b_vals: torch.Tensor, device)
challenges/medium/76_adder_transformer/challenge.py:51
↓ 1 callersFunction_forward_pass
(seq: torch.Tensor, weights: torch.Tensor)
challenges/medium/76_adder_transformer/challenge.py:84
↓ 1 callersMethod_make_rope_tables
(self, seq_len, device, dtype, theta=10000.0)
challenges/hard/93_llama_transformer_block/challenge.py:209
↓ 1 callersMethod_make_weights
(self, device, dtype)
challenges/hard/74_gpt2_block/challenge.py:165
↓ 1 callersMethod_make_weights
(self, device, dtype)
challenges/hard/93_llama_transformer_block/challenge.py:217
↓ 1 callersFunctionextract_id
(name: str)
scripts/update_challenges.py:31
↓ 1 callersFunctionfind_solution_file
(challenge_dir: Path, language: str)
scripts/run_challenge.py:29
↓ 1 callersMethodfnv1a_hash
(self, x: torch.Tensor)
challenges/easy/24_rainbow_table/challenge.py:15
↓ 1 callersFunctionget_backend_filename
(disk_filename: str)
scripts/update_challenges.py:69
↓ 1 callersFunctionget_difficulty
(path: Path)
scripts/update_challenges.py:38
↓ 1 callersFunctionget_language
(filename: str)
scripts/update_challenges.py:46
↓ 1 callersFunctionload_challenge
(problem_dir: Path)
scripts/update_challenges.py:73
↓ 1 callersFunctionmain
()
scripts/update_challenges.py:143
↓ 1 callersFunctionmain
()
scripts/run_challenge.py:90
↓ 1 callersFunctionsubmit_solution
( ws_url: str, api_key: str, challenge_code: str, file_name: str, content: str, langua
scripts/run_challenge.py:47
Method__init__
(self, shape, low=0.0, high=1.0, dtype="float32")
challenges/core/challenge_base.py:8
Method__init__
(self, shape, mean=0.0, std=1.0, dtype="float32")
challenges/core/challenge_base.py:18
Method__init__
(self, shape, low, high, dtype="int32")
challenges/core/challenge_base.py:28
Method__init__
(self, shape, value=0.0, dtype="float32")
challenges/core/challenge_base.py:38
Method__init__
(self, shape, dtype="float32")
challenges/core/challenge_base.py:48
Method__init__
(self, device: str = "cuda")
challenges/core/challenge_base.py:60
Method_make_test
(pairs)
challenges/medium/76_adder_transformer/challenge.py:358
Methodapply_rope
(qk, c, s)
challenges/hard/93_llama_transformer_block/challenge.py:58
Methodbody
(k, d)
challenges/hard/73_all_pairs_shortest_paths/challenge.py:48
Functionclip_kernel
(input, output, lo, hi, N, BLOCK_SIZE: tl.constexpr)
challenges/easy/62_value_clipping/starter/starter.triton.py:7
Functionconv1d_kernel
(input, kernel, output, input_size, kernel_size, BLOCK_SIZE: tl.constexpr)
challenges/easy/9_1d_convolution/starter/starter.triton.py:7
Functionfnv1a_hash
(x: jax.Array)
challenges/easy/24_rainbow_table/starter/starter.jax.py:5
Functionfnv1a_hash
(x)
challenges/easy/24_rainbow_table/starter/starter.triton.py:7
Functionfnv1a_hash
(x: torch.Tensor)
challenges/easy/24_rainbow_table/starter/starter.pytorch.py:4
Functionfnv1a_hash_kernel
(input, output, n_elements, n_rounds, BLOCK_SIZE: tl.constexpr)
challenges/easy/24_rainbow_table/starter/starter.triton.py:21
Functionfnv1a_hash_u32_scalar
(x: cute.Uint32)
challenges/easy/24_rainbow_table/starter/starter.cute.py:5
Methodforward_last
(seq, cur_len)
challenges/medium/76_adder_transformer/challenge.py:236
Functiongeglu
(input, output, N, BLOCK_SIZE: tl.constexpr)
challenges/easy/65_geglu/starter/starter.triton.py:7
Methodgenerate_example_test
(self)
challenges/hard/39_Fast_Fourier_transform/challenge.py:50
Methodgenerate_example_test
(self)
challenges/hard/74_gpt2_block/challenge.py:211
Methodgenerate_example_test
(self)
challenges/hard/73_all_pairs_shortest_paths/challenge.py:63
Methodgenerate_example_test
(self)
challenges/hard/59_sliding_window_attn/challenge.py:61
Methodgenerate_example_test
(self)
challenges/hard/93_llama_transformer_block/challenge.py:260
Methodgenerate_example_test
(self)
challenges/hard/14_multi_agent_sim/challenge.py:80
Methodgenerate_example_test
(self)
challenges/hard/15_sorting/challenge.py:25
Methodgenerate_example_test
(self)
challenges/hard/20_kmeans_clustering/challenge.py:87
Methodgenerate_example_test
(self)
challenges/hard/56_linear_attention/challenge.py:68
Methodgenerate_example_test
(self)
challenges/hard/12_multi_head_attention/challenge.py:70
Methodgenerate_example_test
(self)
challenges/hard/36_radix_sort/challenge.py:34
Methodgenerate_example_test
(self)
challenges/hard/46_bfs_shortest_path/challenge.py:98
Methodgenerate_example_test
(self)
challenges/hard/53_casual_attention/challenge.py:55
Methodgenerate_example_test
Generate an example test case for this problem. Returns: Dictionary with test case parameters
challenges/core/challenge_base.py:81
Methodgenerate_example_test
(self)
challenges/medium/55_attn_w_linear_bias/challenge.py:70
Methodgenerate_example_test
(self)
challenges/medium/78_2d_fft/challenge.py:36
Methodgenerate_example_test
(self)
challenges/medium/35_monte_carlo_integration/challenge.py:39
Methodgenerate_example_test
(self)
challenges/medium/32_int8_quantized_matmul/challenge.py:71
Methodgenerate_example_test
(self)
challenges/medium/67_moe_topk_gating/challenge.py:56
Methodgenerate_example_test
(self)
challenges/medium/80_grouped_query_attention/challenge.py:103
Methodgenerate_example_test
(self)
challenges/medium/82_linear_recurrence/challenge.py:82
Methodgenerate_example_test
(self)
challenges/medium/16_prefix_sum/challenge.py:28
Methodgenerate_example_test
(self)
challenges/medium/72_stream_compaction/challenge.py:33
Methodgenerate_example_test
(self)
challenges/medium/29_top_k_selection/challenge.py:31
Methodgenerate_example_test
(self)
challenges/medium/40_batch_normalization/challenge.py:51
Methodgenerate_example_test
(self)
challenges/medium/60_top_p_sampling/challenge.py:61
Methodgenerate_example_test
(self)
challenges/medium/92_decaying_causal_attention/challenge.py:65
Methodgenerate_example_test
(self)
challenges/medium/69_jacobi_stencil_2d/challenge.py:46
Methodgenerate_example_test
(self)
challenges/medium/48_2d_subarray_sum/challenge.py:47
Methodgenerate_example_test
(self)
challenges/medium/84_swiglu_mlp_block/challenge.py:83
Methodgenerate_example_test
(self)
challenges/medium/43_count_array_element/challenge.py:34
Methodgenerate_example_test
(self)
challenges/medium/17_dot_product/challenge.py:29
Methodgenerate_example_test
(self)
challenges/medium/61_rope_embedding/challenge.py:52
Methodgenerate_example_test
(self)
challenges/medium/37_matrix_power/challenge.py:37
Methodgenerate_example_test
(self)
challenges/medium/25_categorical_cross_entropy_loss/challenge.py:59
Methodgenerate_example_test
(self)
challenges/medium/106_token_embedding_layer/challenge.py:104
Methodgenerate_example_test
(self)
challenges/medium/57_fp16_batched_matmul/challenge.py:38
Methodgenerate_example_test
(self)
challenges/medium/64_weight_dequantization/challenge.py:42
Methodgenerate_example_test
(self)
challenges/medium/75_sparse_matrix_dense_matrix_multiplication/challenge.py:67
Methodgenerate_example_test
(self)
challenges/medium/94_ssm_selective_scan/challenge.py:149
Methodgenerate_example_test
(self)
challenges/medium/28_gaussian_blur/challenge.py:43
Methodgenerate_example_test
(self)
challenges/medium/10_2d_convolution/challenge.py:64
Methodgenerate_example_test
(self)
challenges/medium/51_max_subarray_sum/challenge.py:51
Methodgenerate_example_test
(self)
challenges/medium/81_int4_matmul/challenge.py:108
next →1–100 of 919, ranked by callers