hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix bug: (#3000)

Browse source 
before fix,RAY_FUN_CACHE use only get method ,can only get null
  fix : put after create
This commit is contained in:
bibabolynn 2018-10-03 13:53:54 +08:00 • committed by Robert Nishihara
parent 3ce8eb2d4c
commit 9c606ea06c
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
java/runtime/src/main/java/org/ray/runtime/functionmanager/FunctionManager.java
View file

@ -28,7 +28,7 @@ public class FunctionManager {
* Cache from a RayFunc object to its corresponding FunctionDescriptor. Because
* `LambdaUtils.getSerializedLambda` is expensive.
*/
private static final ThreadLocal<WeakHashMap<Class<RayFunc>, FunctionDescriptor>>
private static final ThreadLocal<WeakHashMap<Class<? extends RayFunc>, FunctionDescriptor>>
RAY_FUNC_CACHE = ThreadLocal.withInitial(WeakHashMap::new);
/**
@ -51,6 +51,7 @@ public class FunctionManager {
final String methodName = serializedLambda.getImplMethodName();
final String typeDescriptor = serializedLambda.getImplMethodSignature();
functionDescriptor = new FunctionDescriptor(className, methodName, typeDescriptor);
RAY_FUNC_CACHE.get().put(func.getClass(),functionDescriptor);
}
return getFunction(driverId, functionDescriptor);
}