Class Case

bench/bench.py:34–218

                -6: Fore.LIGHTBLUE_EX, -7: Fore.BLUE, -8: Fore.MAGENTA }

class Case(object):
    _MAX_ARGS = 4
    _C_BENCH = {0:'prepare_threads', 1:'run_preamble', 2:'magic_broadcast', 
                            3:'npyiter_con', 4:'barrier_init', 5:'task_barrier_final',
                            6:'unlock_global', 7:'run_clean', 8:'release_gil', 
                            9:'reacquire_gil',
                            50: 'cobj_init' }

    _C_THREADS = {150:'thread_init', 200:'thread_tasks', 250:'thread_final'}

    def initZeros(self):
        # return [0.0]*self.tries
        return np.zeros(self.tries, dtype='float64')

    def __init__(self, expr, size, dtype, tries=1, 
                 compares={}, nthreads=ne3.get_nthreads() ):
        self.expr       = expr
        self.size       = size
        self.dtype      = dtype
        self.tries      = tries
        self.compares   = compares    # Should be a dict of {name: function_handle}
        self.nthreads   = nthreads
        self.timings    = defaultdict(self.initZeros, {})
        self.stats      = defaultdict(self.initZeros, {})


    def run(self) -> None:
        """
        """
        # Setup variables x1, x2, x3, x4
        # print( "Running benchmark for '{}' x {} times".format(self.expr, self.tries))
        args = []
        varsDict = vars()
        r = np.empty(self.size, dtype=self.dtype)
        for I in range(Case._MAX_ARGS):
            argName = 'x%d'%I

            if argName in self.expr:
                argArray = np.arange(self.size, dtype=self.dtype)
                varsDict[argName] = argArray
                args.append(argArray)

        x1 = args[0]
        x2 = args[1]
        # Run benchmark `tries` times
        ne3.set_nthreads(self.nthreads)
        ne2.set_num_threads(self.nthreads)
        # Numba does not have dynamic thread setting
        for I in range(self.tries):
            # Grab local references to our attributes
            timings = self.timings; stats = self.stats

            t0 = pc()
            nefunc = ne3.NumExpr( self.expr )
            t1 = pc()
            nefunc()
            t2 = pc()
            timings['total_run'][I] = t2 - t1