Esempio n. 1
0
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
Esempio n. 2
0
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
  
Esempio n. 3
0
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