diff --git a/src/python/ksc/cgen.py b/src/python/ksc/cgen.py index c0171b158..202f04f97 100644 --- a/src/python/ksc/cgen.py +++ b/src/python/ksc/cgen.py @@ -30,7 +30,7 @@ def entry_point_cpp_type(t, use_torch): return scalar_type_to_cpp_map[t.kind] elif t.is_tuple: return ( - "ks::Tuple<" + "std::tuple<" + ", ".join( entry_point_cpp_type(child, use_torch) for child in t.tuple_elems() ) diff --git a/src/runtime/knossos-entry-points.h b/src/runtime/knossos-entry-points.h index 16d4aaec5..004137ffe 100644 --- a/src/runtime/knossos-entry-points.h +++ b/src/runtime/knossos-entry-points.h @@ -42,23 +42,23 @@ struct Converter }; template -struct Converter, ks::Tuple> +struct Converter, std::tuple> { template - static ks::Tuple to_ks_impl(ks::Tuple arg, std::index_sequence) { - return ks::make_Tuple(convert_argument(ks::get(arg))...); + static ks::Tuple to_ks_impl(std::tuple arg, std::index_sequence) { + return ks::make_Tuple(convert_argument(std::get(arg))...); } - static ks::Tuple to_ks(ks::Tuple arg) { + static ks::Tuple to_ks(std::tuple arg) { return to_ks_impl(arg, std::index_sequence_for{}); } template - static ks::Tuple from_ks_impl(ks::Tuple ret, std::index_sequence) { - return ks::make_Tuple(convert_return_value(ks::get(ret))...); + static std::tuple from_ks_impl(ks::Tuple ret, std::index_sequence) { + return std::make_tuple(convert_return_value(ks::get(ret))...); } - static ks::Tuple from_ks(ks::Tuple ret) { + static std::tuple from_ks(ks::Tuple ret) { return from_ks_impl(ret, std::index_sequence_for{}); } };