Skip to content

How to run rools like run_hlo_module and hlo-opt on TPUs? #37277

@SamSz04

Description

@SamSz04

It's feasible to use run_hlo_module and hlo-opt on GPU or CPU through bazel build. I wonder is it possible to use them on Google TPUs? Is there any instruction on this?
Thanks!

Metadata

Metadata

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions