Skip to content

Commit 9c3de1b

Browse files
deqiangctensorflower-gardener
authored andcommitted
Check op_kernel is not null
PiperOrigin-RevId: 689600260
1 parent eba9747 commit 9c3de1b

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

tensorflow/core/tfrt/fallback/op_kernel_runner.cc

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -133,6 +133,11 @@ absl::StatusOr<OpKernelRunner> OpKernelRunner::Create(
133133
std::unique_ptr<OpKernel> op_kernel;
134134
TF_RETURN_IF_ERROR(CreateOpKernel(function_library_runtime,
135135
std::move(node_def), &op_kernel));
136+
137+
if (!op_kernel) {
138+
return absl::InternalError(
139+
absl::StrCat("Failed to create OpKernel for op: ", op_name));
140+
}
136141
return OpKernelRunner(device, function_library_runtime, std::move(op_kernel));
137142
}
138143

0 commit comments

Comments
 (0)