好几次都卡在 test_linalg 上了,跑得好慢,专门跑 python test/test_linalg.py -vv 还有报直接跑太慢了跳过的。
test_linalg_matrix_exp_analytic_cpu_complex128 (__main__.TestLinalgCPU.test_linalg_matrix_exp_analytic_cpu_complex128) ... skipped 'test is slow; run with PYTORCH_TEST_WITH_SLOW to enable test'
test_linalg_matrix_exp_analytic_cpu_complex64 (__main__.TestLinalgCPU.test_linalg_matrix_exp_analytic_cpu_complex64) ... skipped 'test is slow; run with PYTORCH_TEST_WITH_SLOW to enable test'
test_linalg_matrix_exp_analytic_cpu_float32 (__main__.TestLinalgCPU.test_linalg_matrix_exp_analytic_cpu_float32) ... skipped 'test is slow; run with PYTORCH_TEST_WITH_SLOW to enable test'
test_linalg_matrix_exp_analytic_cpu_float64 (__main__.TestLinalgCPU.test_linalg_matrix_exp_analytic_cpu_float64) ... skipped 'test is slow; run with PYTORCH_TEST_WITH_SLOW to enable test'
test_linalg_matrix_exp_batch_cpu_float32 (__main__.TestLinalgCPU.test_linalg_matrix_exp_batch_cpu_float32) ... ok
test_linalg_matrix_exp_batch_cpu_float64 (__main__.TestLinalgCPU.test_linalg_matrix_exp_batch_cpu_float64) ... ok
test_linalg_matrix_exp_boundary_cases_cpu_complex128 (__main__.TestLinalgCPU.test_linalg_matrix_exp_boundary_cases_cpu_complex128) ... ok
test_linalg_matrix_exp_boundary_cases_cpu_complex64 (__main__.TestLinalgCPU.test_linalg_matrix_exp_boundary_cases_cpu_complex64) ... ok
test_linalg_matrix_exp_boundary_cases_cpu_float32 (__main__.TestLinalgCPU.test_linalg_matrix_exp_boundary_cases_cpu_float32) ... ok
test_linalg_matrix_exp_boundary_cases_cpu_float64 (__main__.TestLinalgCPU.test_linalg_matrix_exp_boundary_cases_cpu_float64) ... ok
test_linalg_matrix_exp_compare_with_taylor_cpu_complex128 (__main__.TestLinalgCPU.test_linalg_matrix_exp_compare_with_taylor_cpu_complex128) ... ok
test_linalg_matrix_exp_compare_with_taylor_cpu_complex64 (__main__.TestLinalgCPU.test_linalg_matrix_exp_compare_with_taylor_cpu_complex64) ... ok
test_linalg_matrix_exp_compare_with_taylor_cpu_float32 (__main__.TestLinalgCPU.test_linalg_matrix_exp_compare_with_taylor_cpu_float32) ... ok
test_linalg_matrix_exp_compare_with_taylor_cpu_float64 (__main__.TestLinalgCPU.test_linalg_matrix_exp_compare_with_taylor_cpu_float64) ... ok
test_linalg_matrix_exp_no_warnings_cpu_complex64 (__main__.TestLinalgCPU.test_linalg_matrix_exp_no_warnings_cpu_complex64) ... ok
test_linalg_matrix_exp_perverse_nan_values_cpu_complex128 (__main__.TestLinalgCPU.test_linalg_matrix_exp_perverse_nan_values_cpu_complex128) ...
虽然搜到了一些 openblas 只能设置单线程的 issue,但设置了还是没用, 总是在 test_linalg_matrix_exp_perverse_nan_values_cpu_complex128 单核跑满 100% 不动了。不过 pytorch 的测试集可以设置 @slow 的 tag ,跳过这个测试。
结果又有 setuptools 需要 < 82.0.0