TensorRT-LLMs/tests/unittest/_torch/thop
xiweny 6979afa6f2
test: reorganize tests folder hierarchy (#2996)
1. move TRT path tests to 'trt' folder
2. optimize some import usage
2025-03-27 12:07:53 +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_moe_op.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_noaux_tc.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_scaled_mm.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00
test_selective_scan_op.py test: reorganize tests folder hierarchy (#2996) 2025-03-27 12:07:53 +08:00