PR #6296 Re-land: Fix model initialization.

Re-land: Fix model initialization. #6296

ysiraichi merged 2 commits into master from ysiraichi/fix-model-initialization

Initialize benchmark with accelerator, then move to XLA.

ce21b4be

ysiraichi requested a review from

cota 1 year ago

ysiraichi removed review request from

cota 1 year ago

ysiraichi requested a review from

frgossen 1 year ago

ysiraichi removed review request from

frgossen 1 year ago

ysiraichi requested a review from

cota 1 year ago

ysiraichi requested a review from

golechwierowicz 1 year ago

ysiraichi requested a review from

vanbasten23 1 year ago

ysiraichi added xla:gpu

ysiraichi requested a review from

frgossen 1 year ago

vanbasten23 commented on 2024-01-16

vanbasten23 approved these changes on 2024-01-16

Import pytree before using it.

f77c398c

frgossen approved these changes on 2024-01-16

ysiraichi merged a8b27eb1 into master 1 year ago

Reviewers

frgossen

vanbasten23

cota

golechwierowicz

Assignees

No one assigned

Labels

xla:gpu

Milestone

No milestone