TensorRT-LLMs/tests/functional
2024-11-05 16:27:06 +08:00
..
test_alibi.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +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_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_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 (#1725) 2024-06-04 20:26:32 +08: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 (#1725) 2024-06-04 20:26:32 +08: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_low_latency_gemm.py Update TensorRT-LLM (#2413) 2024-11-05 16:27:06 +08:00
test_mamba_conv1d.py Update TensorRT-LLM (#2008) 2024-07-23 23:05:09 +08: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 (#1725) 2024-06-04 20:26:32 +08:00
test_moe.py Update TensorRT-LLM (#2413) 2024-11-05 16:27:06 +08:00
test_nccl.py Update TensorRT-LLM (#2413) 2024-11-05 16:27:06 +08: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_reduce_norm.py Update TensorRT-LLM (#2413) 2024-11-05 16:27:06 +08:00
test_repeat_interleave.py Update TensorRT-LLM (#1725) 2024-06-04 20:26:32 +08:00
test_rg_lru.py Update TensorRT-LLM (#1688) 2024-05-28 20:07:49 +08: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 (#2094) 2024-08-07 16:44:43 +08: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 (#1725) 2024-06-04 20:26:32 +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 (#2094) 2024-08-07 16:44:43 +08:00