TensorRT-LLMs/tests/functional
Kaiyu Xie 8dd9c91470
Update TensorRT-LLM (#539)
* Update TensorRT-LLM

---------

Co-authored-by: Shixiaowei02 <39303645+Shixiaowei02@users.noreply.github.com>
2023-12-04 18:06:59 +08:00
..
test_alibi.py Update TensorRT-LLM (#506) 2023-11-30 16:46:22 +08:00
test_arange.py Update code 2023-09-28 09:00:05 -07:00
test_assertion.py Initial commit 2023-09-20 00:29:41 -07:00
test_avg_pool2d.py Initial commit 2023-09-20 00:29:41 -07:00
test_cast.py Initial commit 2023-09-20 00:29:41 -07:00
test_conv2d.py Initial commit 2023-09-20 00:29:41 -07:00
test_cos.py Initial commit 2023-09-20 00:29:41 -07:00
test_einsum.py Initial commit 2023-09-20 00:29:41 -07:00
test_embedding_single_gpu.py Initial commit 2023-09-20 00:29:41 -07:00
test_exp.py Initial commit 2023-09-20 00:29:41 -07:00
test_expand.py Update TensorRT-LLM (#539) 2023-12-04 18:06:59 +08:00
test_flip.py Initial commit 2023-09-20 00:29:41 -07:00
test_gather.py Initial commit 2023-09-20 00:29:41 -07:00
test_geglu.py Initial commit 2023-09-20 00:29:41 -07:00
test_gelu.py Update TensorRT-LLM (#506) 2023-11-30 16:46:22 +08:00
test_group_norm.py Initial commit 2023-09-20 00:29:41 -07:00
test_identity.py Initial commit 2023-09-20 00:29:41 -07:00
test_index_select.py Initial commit 2023-09-20 00:29:41 -07:00
test_interpolate.py Initial commit 2023-09-20 00:29:41 -07:00
test_layer_norm.py Initial commit 2023-09-20 00:29:41 -07:00
test_matmul.py Update TensorRT-LLM (#539) 2023-12-04 18:06:59 +08:00
test_nccl.py Update TensorRT-LLM (#539) 2023-12-04 18:06:59 +08:00
test_outer.py Initial commit 2023-09-20 00:29:41 -07:00
test_permute.py Initial commit 2023-09-20 00:29:41 -07:00
test_repeat_interleave.py Update TensorRT-LLM (#506) 2023-11-30 16:46:22 +08:00
test_rms_norm.py Initial commit 2023-09-20 00:29:41 -07:00
test_select.py Initial commit 2023-09-20 00:29:41 -07:00
test_sigmoid.py Initial commit 2023-09-20 00:29:41 -07:00
test_silu.py Initial commit 2023-09-20 00:29:41 -07:00
test_sin.py Initial commit 2023-09-20 00:29:41 -07:00
test_slice.py Update TensorRT-LLM (#539) 2023-12-04 18:06:59 +08:00
test_softplus.py Initial commit 2023-09-20 00:29:41 -07:00
test_split.py Initial commit 2023-09-20 00:29:41 -07:00
test_swiglu.py Initial commit 2023-09-20 00:29:41 -07:00
test_transpose.py Initial commit 2023-09-20 00:29:41 -07:00
test_view.py Update TensorRT-LLM (#539) 2023-12-04 18:06:59 +08:00
torch_ref.py Initial commit 2023-09-20 00:29:41 -07:00