Add `FunctionLibraryRuntime::InstantiateOptions` struct (take 2).
This new struct allows optional arguments to be passed to the `FunctionLibraryRuntime::Instantiate()` API. The new struct is now used to configure the target device for a function instantiation (instead of an attr). This version fixes an issue in the previous attempt whereby identical functions instantiated on different devices in the same task would be canonicalized to the same key and hence receive the same handle in the ProcFLR, but they would only be instantiated locally in one per-device FLR. The fix ensures that the device name is part of the canonicalized form. A test is added to function_test.py that failed with the old version and passed in the new version. END_PUBLIC BEGIN_PUBLIC Automated g4 rollback of changelist 180979141 PiperOrigin-RevId: 181003413
Loading
Please sign in to comment