Skip to content

Commit

Permalink
fix CI
Browse files Browse the repository at this point in the history
  • Loading branch information
frank-dong-ms committed Aug 13, 2024
1 parent b609e17 commit 13ca07f
Show file tree
Hide file tree
Showing 2 changed files with 11 additions and 11 deletions.
6 changes: 3 additions & 3 deletions onnxruntime/core/framework/session_state_utils.cc
Original file line number Diff line number Diff line change
Expand Up @@ -196,7 +196,7 @@ static common::Status DeserializeTensorProto(const Env& env, const std::basic_st
}
}

const common::Status AllocateTensor(
common::Status AllocateTensor(
const onnxruntime::MemBuffer* m,
std::unique_ptr<onnxruntime::Tensor>& p_tensor,
const onnxruntime::DataTypeImpl* const& type,
Expand All @@ -215,7 +215,7 @@ const common::Status AllocateTensor(
return common::Status::OK();
}

const common::Status AllocateTensorOnDeviceOrMemory(
common::Status AllocateTensorOnDeviceOrMemory(
bool use_device_allocator_for_initializers,
onnxruntime::TensorShape& tensor_shape,
const onnxruntime::DataTypeImpl* const& type,
Expand All @@ -234,7 +234,7 @@ const common::Status AllocateTensorOnDeviceOrMemory(
return common::Status::OK();
}

const common::Status CopyTensorFromCPUToDevice(
common::Status CopyTensorFromCPUToDevice(
const onnxruntime::DataTransferManager& data_transfer_mgr,
std::unique_ptr<onnxruntime::Tensor>& p_deserialize_tensor,
std::unique_ptr<onnxruntime::Tensor>& p_tensor,
Expand Down
16 changes: 8 additions & 8 deletions onnxruntime/core/framework/session_state_utils.h
Original file line number Diff line number Diff line change
Expand Up @@ -50,27 +50,27 @@ common::Status SaveInitializedTensors(
const MemoryProfileFunction& memory_profile_func,
std::unordered_map<std::string, std::unique_ptr<Tensor>>& buffered_tensors);

static const common::Status CopyTensorFromCPUToDevice(
const onnxruntime::DataTransferManager& data_transfer_mgr,
std::unique_ptr<onnxruntime::Tensor>& p_deserialize_tensor,
std::unique_ptr<onnxruntime::Tensor>& p_tensor,
OrtValue& ort_value);

static const common::Status AllocateTensor(
common::Status AllocateTensor(
const onnxruntime::MemBuffer* m,
std::unique_ptr<onnxruntime::Tensor>& p_tensor,
const onnxruntime::DataTypeImpl* const& type,
onnxruntime::TensorShape& tensor_shape,
bool use_device_allocator_for_initializers,
const onnxruntime::AllocatorPtr& alloc);

const common::Status AllocateTensorOnDeviceOrMemory(
common::Status AllocateTensorOnDeviceOrMemory(
bool use_device_allocator_for_initializers,
onnxruntime::TensorShape& tensor_shape,
const onnxruntime::DataTypeImpl* const& type,
const onnxruntime::AllocatorPtr& alloc,
std::unique_ptr<onnxruntime::Tensor>& p_tensor);

common::Status CopyTensorFromCPUToDevice(
const onnxruntime::DataTransferManager& data_transfer_mgr,
std::unique_ptr<onnxruntime::Tensor>& p_deserialize_tensor,
std::unique_ptr<onnxruntime::Tensor>& p_tensor,
OrtValue& ort_value);

common::Status SaveInputOutputNamesToNodeMapping(const GraphViewer& graph,
SessionState& session_state,
gsl::span<const NodeArg* const> implicit_inputs);
Expand Down

0 comments on commit 13ca07f

Please sign in to comment.