Provide a CancellationManager to kernels that run in a GraphRunner.
Some kernels are (for example) constant foldable, but use `OpKernelContext::cancellation_manager()` internally (typically when running functions). A recent change made it possible for more kernels to run the GraphRunner (by providing a FunctionLibraryRuntime to the OpKernelConstruction), but no CancellationManager was available, leading to crashes when attempting to run those kernels. PiperOrigin-RevId: 237253404
Loading
Please sign in to comment