| 209 | } |
| 210 | |
| 211 | Status GlProgram::Dispatch(const uint3& workgroups) const { |
| 212 | if (workgroups.x == 0 || workgroups.y == 0 || workgroups.z == 0) { |
| 213 | return InvalidArgumentError("Invalid workgroups"); |
| 214 | } |
| 215 | RETURN_IF_ERROR(TFLITE_GPU_CALL_GL(glUseProgram, id_)); |
| 216 | return TFLITE_GPU_CALL_GL(glDispatchCompute, workgroups.x, workgroups.y, |
| 217 | workgroups.z); |
| 218 | } |
| 219 | |
| 220 | } // namespace gl |
| 221 | } // namespace gpu |
no test coverage detected