Skip to content

Commit b6a836d

Browse files
committed
Fix
1 parent aba7ffd commit b6a836d

File tree

2 files changed

+12
-1
lines changed

2 files changed

+12
-1
lines changed

paddle/fluid/operators/load_combine_op.cc

+11
Original file line numberDiff line numberDiff line change
@@ -81,3 +81,14 @@ namespace ops = paddle::operators; // NOLINT
8181
REGISTER_OPERATOR(load_combine,
8282
ops::LoadCombineOp,
8383
ops::LoadCombineOpProtoMaker);
84+
85+
PD_REGISTER_STRUCT_KERNEL(load_combine_fluid,
86+
CPU,
87+
ALL_LAYOUT,
88+
ops::LoadCombineOpKernel,
89+
float,
90+
double,
91+
phi::dtype::bfloat16,
92+
int,
93+
int8_t,
94+
int64_t) {}

paddle/fluid/operators/ops_signature/load_combine_sig.cc

+1-1
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,7 @@ KernelSignature LoadCombineOpArgumentMapping(
2929
{"file_path", "load_as_fp16", "model_from_memory"},
3030
{"Out"});
3131
} else {
32-
return KernelSignature("load_combine",
32+
return KernelSignature("load_combine_fluid",
3333
{},
3434
{"file_path", "load_as_fp16", "model_from_memory"},
3535
{"Out"});

0 commit comments

Comments
 (0)