TensorRT-LLMs/tests/unittest/_torch/thop
Zhenhuan Chen 2672f13d77
test: fix cublas_scaled_mm with aligned workspace size (#3600)
Signed-off-by: Zhenhuan Chen <chenzhh3671@gmail.com>
2025-04-21 14:51:42 +08:00
..
test_cublas_mm.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_logits_bitmask_op.py Update (#2978) 2025-03-23 16:39:35 +08:00
test_mamba_conv1d_op.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_noaux_tc.py Clean up modeling_deepseek.py (#3640) 2025-04-18 17:54:33 -07:00
test_scaled_mm.py test: fix cublas_scaled_mm with aligned workspace size (#3600) 2025-04-21 14:51:42 +08:00
test_selective_scan_op.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00