Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

move cudnn_lstm kernel to phi #53730

Merged
merged 12 commits into from
May 16, 2023
Merged

move cudnn_lstm kernel to phi #53730

Show file tree
Hide file tree
Changes from 9 commits
Commits
Show all changes
12 commits
Select commit Hold shift + click to select a range
3788d6f
update
huangjiyi May 11, 2023
c70b4b2
fix bug
huangjiyi May 11, 2023
2b0128d
test
huangjiyi May 11, 2023
4da8e3e
test
huangjiyi May 11, 2023
295fc5f
update
huangjiyi May 11, 2023
eac6e10
update mutable_data
huangjiyi May 11, 2023
ea1b7fb
fix bug
huangjiyi May 11, 2023
877b374
update
huangjiyi May 11, 2023
6ce5274
fix bug
huangjiyi May 12, 2023
94362fd
update output type reg
huangjiyi May 12, 2023
de054d2
update
huangjiyi May 12, 2023
055286b
update
huangjiyi May 12, 2023
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 0 additions & 12 deletions paddle/fluid/operators/cudnn_lstm_op.cc
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -292,15 +292,6 @@ class CudnnLSTMGradOpMaker : public framework::SingleGradOpMaker<T> {
}
};

template <typename T, typename DeviceContext>
class NotImpleKernel : public framework::OpKernel<T> {
public:
void Compute(const framework::ExecutionContext& ctx) const override {
PADDLE_THROW(platform::errors::Unimplemented(
"CPU is not support for this kernel now. Will be add in the future"));
}
};

} // namespace operators
} // namespace paddle

Expand All @@ -312,9 +303,6 @@ REGISTER_OPERATOR(cudnn_lstm,
ops::CudnnLSTMGradOpMaker<paddle::imperative::OpBase>);
REGISTER_OPERATOR(cudnn_lstm_grad, ops::CudnnLSTMGradOp);

PD_REGISTER_STRUCT_KERNEL(
cudnn_lstm, CPU, ALL_LAYOUT, ops::NotImpleKernel, float) {}

// TODO(Shixiaowei02) Add ModifyInput support
REGISTER_OP_VERSION(cudnn_lstm)
.AddCheckpoint(
Expand Down
Loading