memoryKernel.hpp

Go to the documentation of this file.

/*
 *   Copyright 2025 Huawei Technologies Co., Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#pragma once
 
#include <acl/acl.h>
#include <hicr/core/exceptions.hpp>
#include <hicr/core/localMemorySlot.hpp>
#include <hicr/backends/ascend/communicationManager.hpp>
#include <hicr/backends/ascend/kernel.hpp>
 
namespace HiCR::backend::ascend
{ 
class MemoryKernel final : public Kernel
{
  public:
 
  MemoryKernel(ascend::CommunicationManager          *commManager,
               std::shared_ptr<HiCR::LocalMemorySlot> destination,
               const size_t                           destinationOffset,
               std::shared_ptr<HiCR::LocalMemorySlot> source,
               const size_t                           sourceOffset,
               size_t                                 size)
    : ascend::Kernel(),
      _dst(destination),
      _src(source),
      _dstOffset(destinationOffset),
      _srcOffset(sourceOffset),
      _size(size),
      _commManager(commManager){};
 
  MemoryKernel() = delete;
 
  ~MemoryKernel() {}
 
  __INLINE__ void start(const aclrtStream stream) override { _commManager->memcpyAsync(_dst, _dstOffset, _src, _srcOffset, _size, stream); }
 
  private:
 
  const std::shared_ptr<HiCR::LocalMemorySlot> _dst;
  const std::shared_ptr<HiCR::LocalMemorySlot> _src;
 
  const size_t _dstOffset;
 
  const size_t _srcOffset;
  const size_t _size;
 
  ascend::CommunicationManager *const _commManager;
};
} // namespace HiCR::backend::ascend