TensorRT-LLMs/tests/unittest/functional
Kaiyu Xie 2631f21089
Update (#2978)
Signed-off-by: Kaiyu Xie <26294424+kaiyux@users.noreply.github.com>
2025-03-23 16:39:35 +08:00
..
test_alibi.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_allreduce_norm.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_allreduce_prepost_residual_norm.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_arange.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_argmax.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_assertion.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_avg_pool2d.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_cast.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_conv2d.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_conv3d.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_cos.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_cumsum.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_dora.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_einsum.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_embedding_single_gpu.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_exp.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_expand.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_flatten.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_flip.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_fp4_gemm_ootb.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_fp4_gemm.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_gather_nd.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_gather.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_geglu.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_gelu.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_gemm_swiglu.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_group_norm.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_identity.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_index_select.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_interpolate.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_logsoftmax.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_lora.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_low_latency_gemm.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_mamba_conv1d.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_masked_scatter.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_masked_select.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_matmul.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_meshgrid2d.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_moe.py Update (#2978) 2025-03-23 16:39:35 +08:00
test_nccl.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_nonzero.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_outer.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_pad.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_permute.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_pp_reduce_scatter.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_quant.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_rearrange.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_repeat_interleave.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_repeat.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_rg_lru.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_sample.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_scatter_nd.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_scatter.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_select.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_selective_scan.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_sigmoid.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_silu.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_sin.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_slice.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_softplus.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_split.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_squeeze.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_swiglu.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_topk.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_transpose.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_unbind.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_unsqueeze.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_view.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
test_where.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00
torch_ref.py Update TensorRT-LLM (#2936) 2025-03-18 21:25:19 +08:00