diff --git a/examples/python/call-external.py b/examples/python/call-external.py index c13d99bd06295096c26d6e113841c853f80645fc..37579fdd8d20c585b7e3db0af079605bb5f46e23 100644 --- a/examples/python/call-external.py +++ b/examples/python/call-external.py @@ -83,12 +83,6 @@ class BLASCallable(lp.ScalarCallable): yield("99_cblas", "#include ") return - -def blas_fn_lookup(target, identifier): - if identifier == 'gemv': - return BLASCallable(name='gemv') - return None - # }}} @@ -105,7 +99,6 @@ knl = lp.make_kernel( target=CTarget(), lang_version=(2018, 2)) -knl = lp.register_function_id_to_in_knl_callable_mapper( - knl, blas_fn_lookup) +knl = lp.register_callable(knl, "gemv", BLASCallable(name="gemv")) print(lp.generate_code_v2(knl).device_code()) diff --git a/examples/python/global_barrier_removal.py b/examples/python/global_barrier_removal.py index be22e268c85fe985a98763426faf8cfadf73c5fb..e09c0d2cb98063f759f6d181db85dfc7b11925ac 100644 --- a/examples/python/global_barrier_removal.py +++ b/examples/python/global_barrier_removal.py @@ -22,7 +22,9 @@ from loopy.preprocess import preprocess_kernel knl = preprocess_kernel(knl) from loopy.schedule import get_one_scheduled_kernel -knl = get_one_scheduled_kernel(knl.root_kernel, knl.callables_table) +knl = knl.with_kernel(get_one_scheduled_kernel(knl["loopy_kernel"], + knl.callables_table)) + # map schedule onto host or device print(knl) diff --git a/examples/python/sparse.py b/examples/python/sparse.py index 7791f41ba808b5fc822277096cf079a5fe6435b5..b4dd07df40007db16ab588c26dfefb4aadb4b7eb 100644 --- a/examples/python/sparse.py +++ b/examples/python/sparse.py @@ -11,9 +11,9 @@ k = lp.make_kernel([ <> length = rowend - rowstart y[i] = sum(j, values[rowstart+j] * x[colindices[rowstart + j]]) end - """) + """, name="spmv") k = lp.add_and_infer_dtypes(k, { - "values,x": np.float64, "rowstarts,colindices": k.root_kernel.index_dtype + "values,x": np.float64, "rowstarts,colindices": k["spmv"].index_dtype }) print(lp.generate_code_v2(k).device_code())