@@ -48,6 +48,8 @@ class DeterminantGradOp : public framework::OperatorWithKernel {
4848 OP_INOUT_CHECK (ctx->HasInput (" Input" ), " Input" , " Input" ,
4949 " DeterminantGradOp" );
5050 OP_INOUT_CHECK (ctx->HasInput (" Out" ), " Input" , " Out" , " DeterminantGradOp" );
51+ OP_INOUT_CHECK (ctx->HasInput (framework::GradVarName (" Out" )), " Input" ,
52+ framework::GradVarName (" Out" ), " DeterminantGradOp" );
5153 OP_INOUT_CHECK (ctx->HasOutput (framework::GradVarName (" Input" )), " Output" ,
5254 framework::GradVarName (" Input" ), " DeterminantGradOp" );
5355
@@ -117,7 +119,8 @@ class SlogDeterminantGradOp : public framework::OperatorWithKernel {
117119 " SlogDeterminantGradOp" );
118120 OP_INOUT_CHECK (ctx->HasInput (" Out" ), " Input" , " Out" ,
119121 " SlogDeterminantGradOp" );
120-
122+ OP_INOUT_CHECK (ctx->HasInput (framework::GradVarName (" Out" )), " Input" ,
123+ framework::GradVarName (" Out" ), " SlogDeterminantGradOp" );
121124 OP_INOUT_CHECK (ctx->HasOutput (framework::GradVarName (" Input" )), " Output" ,
122125 framework::GradVarName (" Input" ), " SlogDeterminantGradOp" );
123126
@@ -179,13 +182,13 @@ REGISTER_OPERATOR(slogdeterminant, ops::SlogDeterminantOp,
179182 ops::SlogDeterminantGradOpMaker<paddle::imperative::OpBase>);
180183
181184REGISTER_OPERATOR (slogdeterminant_grad,
182- ops::DeterminantGradOp ) // reuse det grad op
185+ ops::SlogDeterminantGradOp ) // reuse det grad op
183186
184187REGISTER_OP_CPU_KERNEL(
185188 slogdeterminant, ops::SlogDeterminantKernel<plat::CPUDeviceContext, float >,
186189 ops::SlogDeterminantKernel<plat::CPUDeviceContext, double >);
187190
188191REGISTER_OP_CPU_KERNEL (
189192 slogdeterminant_grad,
190- ops::DeterminantGradKernel <plat::CPUDeviceContext, float >,
191- ops::DeterminantGradKernel <plat::CPUDeviceContext, double >);
193+ ops::SlogDeterminantGradKernel <plat::CPUDeviceContext, float >,
194+ ops::SlogDeterminantGradKernel <plat::CPUDeviceContext, double >);
0 commit comments