Code

Fix a few PyTorch annotations#176

Code

Merged

yuhc merged 2 commits intoutcs-scea:masterutcs-scea/ava:masterfrom anjuna-security:fix/pytorch_annotationanjuna-security/ava:fix/pytorch_annotationCopy head branch name to clipboardJun 17, 2021

+24-8

Commits

Commits on Jun 17, 2021

• Re-enable dynamic cudaMemcpyAsync() memory copy kind in pt_opt

  

  elementgreen

  authored and

  yuhc

  committed
  2cb55a7

• Bug fix in cudnnGetConvolutionForwardAlgorithm_v7 and implemented cudnnGetConvolutionBackwardFilterAlgorithm_v7, cudnnGetConvolutionBackwardDataAlgorithm_v7, and cudaMemsetAsync

  

  elementgreen

  authored and

  yuhc

  committed
  c1e4d54