TensorRT-LLMs/tests/functional
2024-12-24 15:58:43 +08:00
..
test_alibi.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_allreduce_norm.py TensorRT-LLM v0.16 Release 2024-12-24 15:58:43 +08:00
test_arange.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_argmax.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_assertion.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_avg_pool2d.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_cast.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_conv2d.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_cos.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_cumsum.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_einsum.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_embedding_single_gpu.py TensorRT-LLM Release 0.15.0 (#2529) 2024-12-04 13:44:56 +08:00
test_exp.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_expand.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_flatten.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_flip.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_gather_nd.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_gather.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_geglu.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_gelu.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_gemm_swiglu.py TensorRT-LLM Release 0.15.0 (#2529) 2024-12-04 13:44:56 +08:00
test_group_norm.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_identity.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_index_select.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_interpolate.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_logsoftmax.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_low_latency_gemm.py TensorRT-LLM Release 0.15.0 (#2529) 2024-12-04 13:44:56 +08:00
test_mamba_conv1d.py TensorRT-LLM v0.12 Update (#2164) 2024-08-29 17:25:07 +08:00
test_masked_scatter.py Update TensorRT-LLM Release branch (#1445) 2024-04-12 17:59:19 +08:00
test_masked_select.py TensorRT-LLM v0.13 Update (#2269) 2024-09-30 16:20:23 +08:00
test_matmul.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_moe.py TensorRT-LLM Release 0.15.0 (#2529) 2024-12-04 13:44:56 +08:00
test_nccl.py TensorRT-LLM v0.16 Release 2024-12-24 15:58:43 +08:00
test_nonzero.py TensorRT-LLM v0.10 update 2024-06-05 20:43:25 +08:00
test_outer.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_permute.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_pp_reduce_scatter.py TensorRT-LLM Release 0.15.0 (#2529) 2024-12-04 13:44:56 +08:00
test_repeat_interleave.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_rg_lru.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_sample.py TensorRT-LLM v0.10 update 2024-06-05 20:43:25 +08:00
test_scatter_nd.py TensorRT-LLM v0.12 Update (#2164) 2024-08-29 17:25:07 +08:00
test_scatter.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_select.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_selective_scan.py TensorRT-LLM v0.16 Release 2024-12-24 15:58:43 +08:00
test_sigmoid.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_silu.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_sin.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_slice.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_softplus.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_split.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_squeeze.py TensorRT-LLM v0.12 Update (#2164) 2024-08-29 17:25:07 +08:00
test_swiglu.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_topk.py TensorRT-LLM Release 0.15.0 (#2529) 2024-12-04 13:44:56 +08:00
test_transpose.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_unbind.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_unsqueeze.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_view.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
test_where.py TensorRT-LLM v0.11 Update (#1969) 2024-07-17 20:45:02 +08:00
torch_ref.py TensorRT-LLM v0.12 Update (#2164) 2024-08-29 17:25:07 +08:00