Gericom/teak-llvm
1
Fork 0
mirror of https://github.com/Gericom/teak-llvm.git synced 2025-06-19 03:25:54 -04:00
Code Issues Actions Packages Projects Releases Wiki Activity

[libomptarget][nfc] Add declarations of atomic functions for amdgcn

Browse Source 
Summary:
[libomptarget][nfc] Add declarations of atomic functions for amdgcn

This enables building more source for amdgcn. The functions are usually available
in a hip runtime header, but are duplicated here to decouple the implementation

Reviewers: jdoerfert, ABataev, grokos

Reviewed By: jdoerfert

Subscribers: jvesely, mgorny, jfb, openmp-commits

Tags: #openmp

Differential Revision: https://reviews.llvm.org/D71412
This commit is contained in:
Jon Chesterfield 2019-12-12 22:56:13 +00:00
parent 5368f35efa
commit 479868646a
3 changed files with 42 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
openmp/libomptarget/deviceRTLs/amdgcn/CMakeLists.txt
View File

@ -60,6 +60,7 @@ set(cuda_sources
set(h_files
${CMAKE_CURRENT_SOURCE_DIR}/src/amdgcn_interface.h
${CMAKE_CURRENT_SOURCE_DIR}/src/hip_atomics.h
${CMAKE_CURRENT_SOURCE_DIR}/src/target_impl.h
${devicertl_base_directory}/common/debug.h
${devicertl_base_directory}/common/device_environment.h

39
openmp/libomptarget/deviceRTLs/amdgcn/src/hip_atomics.h Normal file
View File

@ -0,0 +1,39 @@
//===---- hip_atomics.h - Declarations of hip atomic functions ---- C++ -*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef OMPTARGET_AMDGCN_HIP_ATOMICS_H
#define OMPTARGET_AMDGCN_HIP_ATOMICS_H
#include "target_impl.h"
DEVICE unsigned atomicAdd(unsigned *address, unsigned val);
DEVICE int atomicAdd(int *address, int val);
DEVICE unsigned long long atomicAdd(unsigned long long *address,
unsigned long long val);
DEVICE unsigned atomicInc(unsigned *address);
DEVICE unsigned atomicInc(unsigned *address, unsigned max);
DEVICE int atomicInc(int *address);
DEVICE int atomicMax(int *address, int val);
DEVICE unsigned atomicMax(unsigned *address, unsigned val);
DEVICE unsigned long long atomicMax(unsigned long long *address,
unsigned long long val);
DEVICE int atomicExch(int *address, int val);
DEVICE unsigned atomicExch(unsigned *address, unsigned val);
DEVICE unsigned long long atomicExch(unsigned long long *address,
unsigned long long val);
DEVICE unsigned atomicCAS(unsigned *address, unsigned compare, unsigned val);
DEVICE int atomicCAS(int *address, int compare, int val);
DEVICE unsigned long long atomicCAS(unsigned long long *address,
unsigned long long compare,
unsigned long long val);
#endif

2
openmp/libomptarget/deviceRTLs/amdgcn/src/target_impl.h
View File

@ -27,6 +27,8 @@
#define SHARED __attribute__((shared))
#define ALIGN(N) __attribute__((aligned(N)))
#include "hip_atomics.h"
////////////////////////////////////////////////////////////////////////////////
// Kernel options
////////////////////////////////////////////////////////////////////////////////