• A
    [cherry-pick] NVIDIA fixes (#43780) · 9edbe4aa
    Aganlengzi 提交于
    * Use all sitepackages path as the library/include path (#42940)
    
    * Fix several unit tests and increase the unit tests stability (#43670)
    
    * Reduce gather op unit tests size and increase the timeout
    
    * Add NVIDIA_TF32_OVERRIDE for multi-processes environment
    
    * Remove record test for device event ut
    
    * Fix 3 unittest errors (#43532)
    
    * Fix test_fuse_resnet_unit failure
    
    * Fix test_imperative_auto_mixed_precision failure
    
    * Fix sparse_attention_op error
    
    * Fix sparse_attention_op error
    
    * Use fixed random seed (#43659)
    
    * for CI test_collective_sendrecv_api
    Co-authored-by: Nzlsh80826 <rewang@nvidia.com>
    Co-authored-by: NShijie <505749828@qq.com>
    9edbe4aa
custom_kernel_dot_setup.py 2.8 KB