Commit c6bb0cf4 authored by gcwenger's avatar gcwenger Committed by Robert Kimball

Eliminated two warnings introduced in #1459 (#1761)

* Eliminated two warnings introduced in #1459

* Removed unnecessary call to reserve_workspace.
parent c16d65c4
......@@ -1065,7 +1065,6 @@ size_t runtime::gpu::CUDNNEmitter::build_primitive(const op::gpu::Rnn* node)
CUDNN_SAFE_CALL(cudnnGetRNNParamsSize(
*m_ctx->cudnn_handle, rnn_desc, temp_input_desc, &params_size, data_type));
auto& w_desc = get_nd_filter_descriptor(Shape{params_size, 1, 1}, data_type, format);
size_t w_idx = allocator.reserve_workspace(params_size);
int num_tensors_per_layer = [&mode] {
switch (mode)
......@@ -1125,8 +1124,6 @@ size_t runtime::gpu::CUDNNEmitter::build_primitive(const op::gpu::Rnn* node)
size_t workspace_idx = allocator.reserve_workspace(workspace_size);
auto recurrent_index = num_tensors_per_layer / 2;
std::unique_ptr<gpu::primitive> kernel_launch(
new gpu::primitive{[=](void** inputs, void** outputs) {
void* workspace_ptr = runtime::gpu::invoke_memory_primitive(m_ctx, workspace_idx);
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment