def run(fn, args):
args = prepare_args(args, fn.input_types)
fn = lower_to_adverbs.apply(fn)
if value_specialization:
fn = specialize(fn, python_values=args)
compiled_fn = CudaCompiler().compile_entry(fn)
assert len(args) == len(fn.input_types)
result = compiled_fn.c_fn(*args)
return result
def run(fn, args):
args = prepare_args(args, fn.input_types)
fn = after_indexify.apply(fn)
# fn = flatten(fn)
fn = final_loop_optimizations.apply(fn)
if value_specialization:
fn = specialize(fn, python_values = args)
compiled_fn = CudaCompiler().compile_entry(fn)
assert len(args) == len(fn.input_types)
result = compiled_fn.c_fn(*args)
return result
def run(fn, args):
args = prepare_args(args, fn.input_types)
fn = lower_to_adverbs.apply(fn)
if value_specialization:
fn = specialize(fn, python_values = args)
compiled_fn = CudaCompiler().compile_entry(fn)
assert len(args) == len(fn.input_types)
result = compiled_fn.c_fn(*args)
return result