30
csrc/aclnn_torch_adapter/NPUBridge.cpp
Normal file
30
csrc/aclnn_torch_adapter/NPUBridge.cpp
Normal file
@@ -0,0 +1,30 @@
|
||||
// Copyright (c) 2020, Huawei Technologies Co., Ltd
|
||||
// All rights reserved.
|
||||
//
|
||||
// This source code is licensed under the BSD-style license found in the
|
||||
// LICENSE file in the root directory of this source tree.
|
||||
|
||||
#include "NPUBridge.h"
|
||||
|
||||
namespace vllm_ascend
|
||||
{
|
||||
NPUStorageImpl *NPUBridge::GetNpuStorageImpl(c10::StorageImpl *storageImpl)
|
||||
{
|
||||
return static_cast<NPUStorageImpl *>(storageImpl);
|
||||
}
|
||||
|
||||
NPUStorageImpl *NPUBridge::GetNpuStorageImpl(c10::Storage &&storage)
|
||||
{
|
||||
return static_cast<NPUStorageImpl *>(storage.unsafeGetStorageImpl());
|
||||
}
|
||||
|
||||
NPUStorageImpl *NPUBridge::GetNpuStorageImpl(const at::Tensor &tensor)
|
||||
{
|
||||
return static_cast<NPUStorageImpl *>(tensor.storage().unsafeGetStorageImpl());
|
||||
}
|
||||
|
||||
NPUStorageDesc &NPUBridge::GetNpuStorageImplDesc(const at::Tensor &tensor)
|
||||
{
|
||||
return static_cast<NPUStorageImpl *>(tensor.storage().unsafeGetStorageImpl())->npu_desc_;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user