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

Explicitly require OpenGL ES 3.1 to use OpenGL inference.

Browse Source 
PiperOrigin-RevId: 253908072
This commit is contained in:
A. Unique TensorFlower 2019-06-18 18:06:04 -07:00 committed by TensorFlower Gardener
parent a41f003d37
commit e982459f83
1 changed files with 13 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tensorflow/lite/delegates/gpu/gl/api.cc
View File

@ -380,6 +380,11 @@ bool IsBatchMatchesForAllValues(const GraphFloat32& model) {
return true; return true;
} }
bool IsOpenGl31OrAbove(const GpuInfo& gpu_info) {
return (gpu_info.major_version == 3 && gpu_info.minor_version >= 1) ||
gpu_info.major_version > 3;
}
} // namespace } // namespace
Status Compile(const CompilationOptions& options, const GraphFloat32& model, Status Compile(const CompilationOptions& options, const GraphFloat32& model,
@ -392,6 +397,10 @@ Status Compile(const CompilationOptions& options, const GraphFloat32& model,
} }
GpuInfo gpu_info; GpuInfo gpu_info;
RETURN_IF_ERROR(RequestGpuInfo(&gpu_info)); RETURN_IF_ERROR(RequestGpuInfo(&gpu_info));
if (!IsOpenGl31OrAbove(gpu_info)) {
return InternalError(
"OpenGL ES 3.1 or above is required to use OpenGL inference.");
}
auto compiled_model_impl = absl::make_unique<CompiledModelImpl>(gpu_info); auto compiled_model_impl = absl::make_unique<CompiledModelImpl>(gpu_info);
compiled_model_impl->set_dynamic_batch(options.dynamic_batch); compiled_model_impl->set_dynamic_batch(options.dynamic_batch);
auto compiler = NewCompiler(&node_shader, &gpu_info, options); auto compiler = NewCompiler(&node_shader, &gpu_info, options);
@ -408,6 +417,10 @@ Status ReadSerializedModel(const std::vector<uint8_t>& serialized_model,
std::unique_ptr<CompiledModel>* compiled_model) { std::unique_ptr<CompiledModel>* compiled_model) {
GpuInfo gpu_info; GpuInfo gpu_info;
RETURN_IF_ERROR(RequestGpuInfo(&gpu_info)); RETURN_IF_ERROR(RequestGpuInfo(&gpu_info));
if (!IsOpenGl31OrAbove(gpu_info)) {
return InternalError(
"OpenGL ES 3.1 or above is required to use OpenGL inference.");
}
auto compiled_model_impl = absl::make_unique<CompiledModelImpl>(gpu_info); auto compiled_model_impl = absl::make_unique<CompiledModelImpl>(gpu_info);
RETURN_IF_ERROR(DeserializeCompiledModel( RETURN_IF_ERROR(DeserializeCompiledModel(
absl::MakeConstSpan(serialized_model), compiled_model_impl.get())); absl::MakeConstSpan(serialized_model), compiled_model_impl.get()));