Program Listing for File OpenMPTargetMemsetOperation.cpp

↰ Return to documentation for file (umpire/op/OpenMPTargetMemsetOperation.cpp)

//////////////////////////////////////////////////////////////////////////////
// Copyright (c) 2016-20, Lawrence Livermore National Security, LLC and Umpire
// project contributors. See the COPYRIGHT file for details.
//
// SPDX-License-Identifier: (MIT)
//////////////////////////////////////////////////////////////////////////////
#include "umpire/op/OpenMPTargetMemsetOperation.hpp"

#include <cstring>

#include "umpire/strategy/AllocationStrategy.hpp"
#include "umpire/util/Macros.hpp"

#include "omp.h"

namespace umpire {
namespace op {

void OpenMPTargetMemsetOperation::apply(
    void* src_ptr,
    util::AllocationRecord* src_allocation,
    int val,
    std::size_t length)
{
  int device = src_allocation->strategy->getTraits().id;
  unsigned char* data_ptr{static_cast<unsigned char*>(src_ptr)};

#pragma omp target is_device_ptr(data_ptr) device(device)
#pragma omp teams distribute parallel for schedule(static, 1)
  for (std::size_t i = 0; i < length; ++i ) {
    data_ptr[i] = static_cast<unsigned char>(val);
  }

  UMPIRE_RECORD_STATISTIC(
      "OpenMPTargetMemsetOperation",
      "src_ptr", reinterpret_cast<uintptr_t>(src_ptr),
      "value", value,
      "size", length,
      "event", "memset");
}

} // end of namespace op
} // end of namespace umpire