Skip to content

Conversation

bghimireamd
Copy link
Contributor

  • add CK's Batch Norm Forward Training solver for F16, F32, F64, BF16 types
  • NHWC Layout
  • GTest added

@bghimireamd bghimireamd changed the title bg/LWPMIOPEN-192: add batch norm foward CK kernel bg/LWPMIOPEN-192: Integrate CK's batch norm forward training into non-tunable MIOpen solver Sep 11, 2023
@bghimireamd bghimireamd added this to the ROCm 6.0 milestone Sep 11, 2023
@junliume
Copy link
Contributor

@bghimireamd the following test is not passing

 test_bn_spatial_nhwc_test

@CAHEK7
Copy link
Contributor

CAHEK7 commented Sep 14, 2023

Could you please apply #2379

@bghimireamd bghimireamd changed the base branch from bg/LWPMIOPEN-194 to bg/LWPMIOPEN-193_bn_back September 27, 2023 21:11
@junliume junliume merged commit d5eb31c into bg/LWPMIOPEN-193_bn_back Oct 5, 2023
@JehandadKhan JehandadKhan deleted the bg/LWPMIOPEN-192 branch December 13, 2023 17:21
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants