xla
Enable cross entropy loss for xla autocast with FP32 precision (#7992)
#8094

Merged

Enable cross entropy loss for xla autocast with FP32 precision (#7992) #8094

lsy323 merged 1 commit into master from enable_autocast_cel

Enable cross entropy loss for xla autocast with FP32 precision (#7992)

b21d2c04

jeffhataws requested a review from

lsy323 1 year ago

jeffhataws requested a review from

cowanmeg 1 year ago

jeffhataws requested a review from

JackCaoG 1 year ago

lsy323 approved these changes on 2024-09-30

lsy323 merged 940bee45 into master 1 year ago

lsy323 deleted the enable_autocast_cel branch 1 year ago

lsy323 restored the head branch 1 year ago

lsy323 deleted the enable_autocast_cel branch 1 year ago

Reviewers

lsy323

cowanmeg

JackCaoG

Assignees

No one assigned

Labels

None yet

Milestone

No milestone