experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix API initializer bug: request TPU library initialize itself during loading

Browse Source 
PiperOrigin-RevId: 347112603
Change-Id: I3951680b9c7873e507363dd40da49da9e9764b4c
This commit is contained in:
Frank Chen 2020-12-11 18:20:04 -08:00 committed by TensorFlower Gardener
parent 41048da313
commit 4b4893e25b
1 changed files with 1 additions and 1 deletions

2
tensorflow/core/tpu/tpu_api_dlsym_initializer.cc
View File

@ -52,7 +52,7 @@ Status InitializeTpuLibrary(void* library_handle) {
void (*initialize_fn)(bool init_library, int argc, char** argv);
initialize_fn = reinterpret_cast<decltype(initialize_fn)>(
dlsym(library_handle, "TfTpu_Initialize"));
(*initialize_fn)(/*init_library=*/false, /*argc=*/0, /*argv=*/nullptr);
(*initialize_fn)(/*init_library=*/true, /*argc=*/0, /*argv=*/nullptr);
RegisterTpuPlatform();
RegisterTpuSystemDevice();