d233e787f0
This reverts commit 7675fc79c802cf7f6a95660f6ee59bf6cb62102f. Requested in PR: https://github.com/llvm/llvm-project/pull/169638#issuecomment-3634227707
84 lines
1.8 KiB
Fortran
84 lines
1.8 KiB
Fortran
!===-- module/cooperative_groups.f90 ---------------------------------------===!
|
|
!
|
|
! 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
|
|
!
|
|
!===------------------------------------------------------------------------===!
|
|
|
|
! CUDA Fortran cooperative groups
|
|
|
|
module cooperative_groups
|
|
|
|
use, intrinsic :: __fortran_builtins, only: c_devptr => __builtin_c_devptr
|
|
|
|
implicit none
|
|
|
|
type :: cluster_group
|
|
type(c_devptr), private :: handle
|
|
integer(4) :: size
|
|
integer(4) :: rank
|
|
end type cluster_group
|
|
|
|
type :: grid_group
|
|
type(c_devptr), private :: handle
|
|
integer(4) :: size
|
|
integer(4) :: rank
|
|
end type grid_group
|
|
|
|
type :: coalesced_group
|
|
type(c_devptr), private :: handle
|
|
integer(4) :: size
|
|
integer(4) :: rank
|
|
end type coalesced_group
|
|
|
|
type :: thread_group
|
|
type(c_devptr), private :: handle
|
|
integer(4) :: size
|
|
integer(4) :: rank
|
|
end type thread_group
|
|
|
|
interface
|
|
attributes(device) function cluster_block_index()
|
|
import
|
|
type(dim3) :: cluster_block_index
|
|
end function
|
|
end interface
|
|
|
|
interface
|
|
attributes(device) function cluster_dim_blocks()
|
|
import
|
|
type(dim3) :: cluster_dim_blocks
|
|
end function
|
|
end interface
|
|
|
|
interface
|
|
attributes(device) function this_cluster()
|
|
import
|
|
type(cluster_group) :: this_cluster
|
|
end function
|
|
end interface
|
|
|
|
interface
|
|
attributes(device) function this_grid()
|
|
import
|
|
type(grid_group) :: this_grid
|
|
end function
|
|
end interface
|
|
|
|
interface
|
|
attributes(device) function this_thread_block()
|
|
import
|
|
type(thread_group) :: this_thread_block
|
|
end function
|
|
end interface
|
|
|
|
interface this_warp
|
|
attributes(device) function this_warp()
|
|
import
|
|
type(coalesced_group) :: this_warp
|
|
end function
|
|
end interface
|
|
|
|
end module
|