Function STRIDED_LOOP

code_generators/interp_generator.py:183–197

(expr: str)

        }'''.replace('EXPR', expr) 
 
def STRIDED_LOOP(expr: str) -> str:
    expr = expr.replace( '$ARG3', ARG_STRIDE(3))
    expr = expr.replace( '$ARG2', ARG_STRIDE(2))
    expr = expr.replace( '$ARG1', ARG_STRIDE(1))
    retStr = ''
    if 'x3' in expr: 
        retStr += '    sb3 /= sizeof($DTYPE3);\n'
    if 'x2' in expr: 
        retStr += '    sb2 /= sizeof($DTYPE2);\n'
    if 'x1' in expr: 
        retStr += '    sb1 /= sizeof($DTYPE1);\n'
    retStr += ''&#x27;    for(npy_intp J = 0; J < task_size; J++) { 
        EXPR; 
    }'''.replace('EXPR&#x27;, expr) 
    return retStr


def VEC_ARG0(expr: str) -> str: