Skip to content

Commit

Permalink
Fix
Browse files Browse the repository at this point in the history
  • Loading branch information
olpipi committed Jan 3, 2024
1 parent cecdbb5 commit a70cd64
Show file tree
Hide file tree
Showing 4 changed files with 15 additions and 8 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,7 @@
#include <vector>

#include "common_test_utils/node_builders/eltwise.hpp"
#include "ov_models/utils/data_utils.hpp"

namespace LayerTestsDefinitions {

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -12,6 +12,7 @@
#include "cuda/device_pointers.hpp"
#include "cuda_operation_registry.hpp"
#include "cuda_thread_context.hpp"
#include "openvino/op/constant.hpp"
#include "unsymmetrical_comparer.hpp"

namespace LayerTestsDefinitions {
Expand Down Expand Up @@ -180,11 +181,11 @@ class CUDALSTMSequenceOptimizedTest : public testing::WithParamInterface<LSTMSeq
seq_lengths_node->set_friendly_name("seq_lengths");
params.push_back(param);
} else if (m_mode == SequenceTestsMode::PURE_SEQ_RAND_SEQ_LEN_CONST) {
seq_lengths_node = ov::test::utils::make_constant<int64_t>(
seq_lengths_node = ov::test::utils::deprecated::make_constant<int64_t>(
ov::element::i64, input_shapes[3], {}, true, static_cast<int64_t>(seq_lengths), 0.f);
} else {
std::vector<int64_t> lengths(input_shapes[3][0], seq_lengths);
seq_lengths_node = ov::test::utils::make_constant(ov::element::i64, input_shapes[3], lengths, false);
seq_lengths_node = ov::op::v0::Constant::create(ov::element::i64, input_shapes[3], lengths);
}
std::shared_ptr<ov::Node> W, R, B;
if (WRBType == InputLayerType::PARAMETER) {
Expand All @@ -198,9 +199,9 @@ class CUDALSTMSequenceOptimizedTest : public testing::WithParamInterface<LSTMSeq
params.push_back(R_param);
params.push_back(B_param);
} else {
W = ov::test::utils::make_constant(ngPrc, W_shape, {}, true, up_to, start_from, counter++);
R = ov::test::utils::make_constant(ngPrc, R_shape, {}, true, up_to, start_from, counter++);
B = ov::test::utils::make_constant(ngPrc, B_shape, {}, true, up_to, start_from, counter++);
W = ov::test::utils::deprecated::make_constant(ngPrc, W_shape, {}, true, up_to, start_from, counter++);
R = ov::test::utils::deprecated::make_constant(ngPrc, R_shape, {}, true, up_to, start_from, counter++);
B = ov::test::utils::deprecated::make_constant(ngPrc, B_shape, {}, true, up_to, start_from, counter++);
}

auto lstm_sequence = std::make_shared<ov::op::v5::LSTMSequence>(x_node,
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -11,12 +11,13 @@
#include <cuda_op_buffers_extractor.hpp>
#include <cuda_operation_registry.hpp>
#include <cuda_simple_execution_delegator.hpp>
#include <ngraph/node.hpp>
#include <kernels/details/cuda_type_traits.hpp>
#include <openvino/op/constant.hpp>
#include <openvino/op/parameter.hpp>
#include <openvino/op/range.hpp>
#include <openvino/op/result.hpp>
#include <ops/parameter.hpp>

#include "kernels/details/cuda_type_traits.hpp"

namespace {

using devptr_t = CUDA::DevicePointer<void*>;
Expand Down
4 changes: 4 additions & 0 deletions modules/nvidia_plugin/tests/unit/cuda_multi_graph_test.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,10 @@
#include "common_test_utils/node_builders/eltwise.hpp"
#include "cuda_graph_topology_runner.hpp"
#include "cuda_simple_execution_delegator.hpp"
#include "openvino/op/concat.hpp"
#include "openvino/op/constant.hpp"
#include "openvino/op/parameter.hpp"
#include "openvino/op/result.hpp"
#include "ops/parameter.hpp"
#include "ops/result.hpp"
#include "ov_models/utils/data_utils.hpp"
Expand Down

0 comments on commit a70cd64

Please sign in to comment.