onnxruntime/orttraining
Vincent Wang 807b2f4dd5
[ORTModule] Use Env Variable to Set Provider Option cudnn_conv_algo_search (#13296)
This PR is to add support of using env variable to set provider option
cudnn_conv_algo_search so that user can choose better conv algo search
method to run model. This is a quick fix to unblock the test of MoE
model. Will have another PR to design and implement the ORTModule config
so that we can config ORTModule using Python script or config file
instead of env variable.
2022-10-13 15:36:21 +08:00
..
orttraining [ORTModule] Use Env Variable to Set Provider Option cudnn_conv_algo_search (#13296) 2022-10-13 15:36:21 +08:00
pytorch_frontend_examples Set black's target version (#11370) 2022-04-27 14:52:19 -07:00
tools [ROCm] Revert CI pipeline to ROCm5.2.3 (#13297) 2022-10-12 10:47:33 -07:00