experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
104,463 Commits 49 Branches 133 Tags 692 MiB
mm-llvm-bazel
Commit Graph

10 Commits

Author SHA1 Message Date
Wenhao Jia
597741a69d Add a TpuTopologyExternal::CoreForId method. 
PiperOrigin-RevId: 354581402
Change-Id: I7dc8f1b987e9f90a4f7d2d751de573f8b01eeb70
 2021-01-29 12:05:55 -08:00
A. Unique TensorFlower
d1d4c51fcd Add HostCount() / ChipsPerHost() API to TPUTopologyExternal. 
PiperOrigin-RevId: 335962331
Change-Id: I4d9a3f8770039348dfc77616333d80c68f493bda
 2020-10-07 15:39:48 -07:00
David Majnemer
4267a5b239 Internal change 
PiperOrigin-RevId: 330448092
Change-Id: Ia1d108fcff63ce4f65c9a219ad30c85754e54c05
 2020-09-07 22:25:33 -07:00
Skye Wanderman-Milne
565f670d9f Add TpuHostLocationExternal::Cores() and C API plumbing. 
PiperOrigin-RevId: 329997425
Change-Id: Ib89ab30c1b01627edc5ca2dac471f622891b099e
 2020-09-03 14:51:14 -07:00
Skye Wanderman-Milne
6a2f00362e Add TpuTopologyExternal::version() and TpuVersionEnumToString(). 
PiperOrigin-RevId: 326040796
Change-Id: I3033d21271db465e4d6c57f5a3fe3c0c8913d1aa
 2020-08-11 09:58:22 -07:00
Skye Wanderman-Milne
82c043fee5 Create dummy topology types for libtpu C API. 
This gives us better typechecking than just using void*.

PiperOrigin-RevId: 325919251
Change-Id: Idc375ed151a2b0b2cf7d62b7be9cf2fafe48e934
 2020-08-10 17:13:50 -07:00
Skye Wanderman-Milne
f7ca4c921e Add TpuTopologyExternal::cores(). 
Also expose the underlying TpuCoreLocationExternal pointer so
internal users can get at not-yet-exposed functionality.

PiperOrigin-RevId: 325855758
Change-Id: I640e4d4996f60689cd533321434a5fec5f612c60
 2020-08-10 12:02:08 -07:00
Skye Wanderman-Milne
c5543bc170 Add TpuCoreLocationExternal::host_coordinates() and TpuTopologyExternal::IdForHost(). 
This also refactors TpuCoreLocationExternal::chip_coordinates() and
downstream plumbing to be more compact.

PiperOrigin-RevId: 325135180
Change-Id: Ic350d3d64b01fced4d6d90551f71e97b271e0f2b
 2020-08-05 17:32:22 -07:00
Skye Wanderman-Milne
cf2e284a9a Add host location plumbing to TPU client. 
This change adds a TpuHostLocationExternal class, which currently only
supports getting the host's ID, and a getter method on TpuPlatformIn
terface (the host location is fetched from the platform, instead of
the topology, since it varies across each host).

PiperOrigin-RevId: 324281861
Change-Id: I30704a480fa5f801161c12b67b5db6bdf690e858
 2020-07-31 14:41:48 -07:00
Frank Chen
44ce8c4417 Introduce public version of TpuTopology for chip/core location identification 
PiperOrigin-RevId: 321644994
Change-Id: I5def78e34c127c1a16ee64da13e83590376b1622
 2020-07-16 14:28:02 -07:00