Skip to content
3 changes: 2 additions & 1 deletion src/libspu/core/context.h
Original file line number Diff line number Diff line change
Expand Up @@ -107,7 +107,8 @@ class KernelEvalContext final {
Index, //
Strides, //
Sizes, //
FieldType // field type for dynamic ring
FieldType, // field type for dynamic ring
double
>;

SPUContext* sctx_;
Expand Down
33 changes: 33 additions & 0 deletions src/libspu/kernel/hal/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -314,3 +314,36 @@ spu_cc_library(
":ring",
],
)


spu_cc_library(
name = "join",
srcs = ["join.cc"],
hdrs = ["join.h"],
deps = [
":type_cast",
"//libspu/core:context",
"//libspu/core:value",
":polymorphic",
":public_helper",
":ring",
":shape_ops",
"@yacl//yacl/utils:cuckoo_index",
"//libspu/kernel/hal:permute",
"//libspu/kernel/hal:utils",
"//libspu/kernel/hal:soprf",
],
)

spu_cc_test(
name = "join_test",
srcs = ["join_test.cc"],
deps = [
"//libspu/kernel:test_util",
"//libspu/mpc/utils:simulate",
"//libspu/kernel/hal:join",
"//libspu/kernel/hlo:casting",
"//libspu/kernel/hlo:geometrical",
"//libspu/kernel/hlo:const",
],
)
Loading