no barrier time

Author: Ehsan Totoni

hpat/__init__.py

@@ -3,6 +3,8 @@
 import numba
 from numba import *
 import hpat.dict_ext
+import hpat.distributed_api
+from hpat.distributed_api import dist_time
 from hpat.dict_ext import DictIntInt, DictInt32Int32, dict_int_int_type, dict_int32_int32_type
 import hpat.str_ext
 

hpat/_distributed.c

@@ -10,6 +10,7 @@ int64_t hpat_dist_get_start(int64_t total, int num_pes, int node_id);
 int64_t hpat_dist_get_end(int64_t total, int num_pes, int node_id);
 int64_t hpat_dist_get_node_portion(int64_t total, int num_pes, int node_id);
 double hpat_dist_get_time();
+double hpat_get_time();
 MPI_Datatype get_MPI_typ(int typ_enum);
 int get_elem_size(int type_enum);
 int hpat_dist_reduce_i4(int value);
@@ -52,6 +53,8 @@ PyMODINIT_FUNC PyInit_hdist(void) {
                             PyLong_FromVoidPtr((void*)(&hpat_dist_get_node_portion)));
     PyObject_SetAttrString(m, "hpat_dist_get_time",
                             PyLong_FromVoidPtr((void*)(&hpat_dist_get_time)));
+    PyObject_SetAttrString(m, "hpat_get_time",
+                            PyLong_FromVoidPtr((void*)(&hpat_get_time)));
 
     PyObject_SetAttrString(m, "hpat_dist_reduce_i4",
                             PyLong_FromVoidPtr((void*)(&hpat_dist_reduce_i4)));
@@ -140,6 +143,12 @@ double hpat_dist_get_time()
     return wtime;
 }
 
+double hpat_get_time()
+{
+    double wtime;
+    wtime = MPI_Wtime();
+    return wtime;
+}
 
 int hpat_dist_reduce_i4(int value)
 {

hpat/distributed_api.py

@@ -46,6 +46,9 @@ def dist_exscan(value):
 def dist_setitem(arr, index, val):
     return 0
 
+def dist_time():
+    return time.time()
+
 def irecv():
     return 0
 
@@ -118,6 +121,13 @@ def generic(self, args, kws):
         assert len(args)==0
         return signature(types.float64, *args)
 
+@infer_global(dist_time)
+class DistDistTime(AbstractTemplate):
+    def generic(self, args, kws):
+        assert not kws
+        assert len(args)==0
+        return signature(types.float64, *args)
+
 @infer_global(dist_cumsum)
 @infer_global(dist_cumprod)
 class DistCumsumprod(AbstractTemplate):

hpat/distributed_lower.py

@@ -15,6 +15,7 @@
 ll.add_symbol('hpat_dist_get_end', hdist.hpat_dist_get_end)
 ll.add_symbol('hpat_dist_get_node_portion', hdist.hpat_dist_get_node_portion)
 ll.add_symbol('hpat_dist_get_time', hdist.hpat_dist_get_time)
+ll.add_symbol('hpat_get_time', hdist.hpat_get_time)
 ll.add_symbol('hpat_dist_reduce_i4', hdist.hpat_dist_reduce_i4)
 ll.add_symbol('hpat_dist_reduce_i8', hdist.hpat_dist_reduce_i8)
 ll.add_symbol('hpat_dist_reduce_f4', hdist.hpat_dist_reduce_f4)
@@ -101,6 +102,12 @@ def lower_dist_arr_reduce(context, builder, sig, args):
 
 @lower_builtin(time.time)
 def dist_get_time(context, builder, sig, args):
+    fnty = lir.FunctionType(lir.DoubleType(), [])
+    fn = builder.module.get_or_insert_function(fnty, name="hpat_get_time")
+    return builder.call(fn, [])
+
+@lower_builtin(distributed_api.dist_time)
+def dist_get_dist_time(context, builder, sig, args):
     fnty = lir.FunctionType(lir.DoubleType(), [])
     fn = builder.module.get_or_insert_function(fnty, name="hpat_dist_get_time")
     return builder.call(fn, [])