added env (CK_TARGET_PATH) to program to get an access to machine spe…

…cific files (to support ReQuEST AI benchmark submission with Apache AVRO)
ctuning · May 3, 2018 · aeb87ef · aeb87ef
1 parent b4f5992
commit aeb87ef
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/CHANGES b/CHANGES
@@ -1,3 +1,6 @@
+* 2018.05.03 - added env (CK_TARGET_PATH) to program to get an access to machine specific files
+               (to support ReQuEST AI benchmark submission with Apache AVRO)
+
 * 2018.04.07 - added support for RPC remote machine (ck add machine) - useful for accessing TVM/MXNET
 
 * 2018.03.09 - improved cross-compilation for Windows for programs

diff --git a/module/program/module.py b/module/program/module.py
@@ -505,13 +505,18 @@ def process_in_dir(i):
 
     # Update env for host from host/target OS desc if needed (for example for RPC)
     x=hosd.get('preset_host_env',{})
-   
+
     if len(x)>0:
        env.update(x)
     x=tosd.get('preset_host_env',{})
     if len(x)>0:
        env.update(x)
 
+    # Add path to CK target entry if used (to get machine specific files if needed)
+    x=device_cfg.get('path_to_ck_target_entry','')
+    if x!='':
+       env['CK_TARGET_PATH']=x
+
     # update misc
     misc['host_os_uoa']=hosx
     misc['target_os_uoa']=tosx