wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-30 07:20:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

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
Download Patch File Download Diff File Expand all files Collapse all files
java/runtime/src/main/java/org/ray/runtime/functionmanager/FunctionManager.java
+2 -1
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);
}