TensorRT-LLMs/tests/functional
2025-02-13 18:40:22 +08:00
..
test_alibi.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_allreduce_norm.py Update TensorRT-LLM (#2532) 2024-12-04 21:16:56 +08:00
test_allreduce_prepost_residual_norm.py Update TensorRT-LLM (#2582) 2024-12-16 21:50:47 -08:00
test_arange.py Update TensorRT-LLM (#1793) 2024-06-18 18:18:23 +08:00
test_argmax.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_assertion.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_avg_pool2d.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_cast.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_conv2d.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_cos.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_cumsum.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_dora.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
test_einsum.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_embedding_single_gpu.py Update TensorRT-LLM (#2389) 2024-10-29 22:24:38 +08:00
test_exp.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_expand.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_flatten.py Update TensorRT-LLM (#1688) 2024-05-28 20:07:49 +08:00
test_flip.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_fp4_gemm_ootb.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
test_fp4_gemm.py Update TensorRT-LLM (#2783) 2025-02-13 18:40:22 +08:00
test_gather_nd.py Update TensorRT-LLM (#1793) 2024-06-18 18:18:23 +08:00
test_gather.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_geglu.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_gelu.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
test_gemm_swiglu.py Update TensorRT-LLM (#2413) 2024-11-05 16:27:06 +08:00
test_group_norm.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_identity.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
test_index_select.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_interpolate.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_logsoftmax.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_lora.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
test_low_latency_gemm.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
test_mamba_conv1d.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
test_masked_scatter.py Update TensorRT-LLM (#1274) 2024-03-12 18:15:52 +08:00
test_masked_select.py Update TensorRT-LLM (#2184) 2024-09-03 12:14:23 +02:00
test_matmul.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
test_moe.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
test_nccl.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
test_nonzero.py Update TensorRT-LLM (#1492) 2024-04-24 14:44:22 +08:00
test_outer.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_permute.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_pp_reduce_scatter.py Update TensorRT-LLM (#2413) 2024-11-05 16:27:06 +08:00
test_quant.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
test_repeat_interleave.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_rg_lru.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
test_sample.py Update TensorRT-LLM (#1492) 2024-04-24 14:44:22 +08:00
test_scatter_nd.py Update TensorRT-LLM (#1891) 2024-07-04 14:37:19 +08:00
test_scatter.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_select.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_selective_scan.py Update TensorRT-LLM (#2755) 2025-02-11 03:01:00 +00:00
test_sigmoid.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_silu.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_sin.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_slice.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_softplus.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_split.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_squeeze.py Update TensorRT-LLM (#1954) 2024-07-16 15:30:25 +08:00
test_swiglu.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_topk.py Update TensorRT-LLM (#2436) 2024-11-12 15:27:49 +08:00
test_transpose.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_unbind.py Update TensorRT-LLM (#1688) 2024-05-28 20:07:49 +08:00
test_unsqueeze.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_view.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_where.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
torch_ref.py Update TensorRT-LLM (#2582) 2024-12-16 21:50:47 -08:00