Skip to content

Commit

Permalink
[MODIFIED]: Changing the grad kernel into atomicAdd style, which woul…
Browse files Browse the repository at this point in the history 
…d inevitably increase thread-safety once calcalating the backward step of index_sample OP, and one special CUDA kernel considering the condition that each line of index array only contains 1 element. Besides, thread-deployment in block was 2-demensions.
  • Loading branch information
@JamesLim-sy
JamesLim-sy committed Jan 22, 2021
2 parents 98a9af7 + d18b4bf commit fec47c5
Showing 0 changed files with 0 additions and 0 deletions.

0 comments on commit fec47c5

Please sign in to comment.