Skip to content

Conversation

@conglongli
Copy link
Contributor

@conglongli conglongli commented Feb 18, 2021

Rollback some changes from #661. This PR fix the error that when validation is enabled ('--validation_data_path_prefix /data/bert ' in bert pretrain example), it would trigger a RuntimeError: CUDA error: an illegal memory access was encountered

@conglongli conglongli changed the title Fix transformer kernel to avoid CUDA illegal memory access error when validation is enabled Fix transformer kernel CUDA illegal memory access error Feb 18, 2021
@conglongli conglongli merged commit 1fcc5f7 into master Feb 18, 2021
@conglongli conglongli deleted the conglli/kernel-patch branch February 18, 2021 07:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants