0

我正在尝试在驻留在 GPU 内存中的“设备”数据结构中使用可分配数组。代码（粘贴在下面）编译，但给出了段错误。我在做一些明显错误的事情吗？

模块文件称为“gpu_modules.F90”，如下所示：

!=============
! This module contains definitions for data structures and the data
! stored on the device
!=============

   module GPU_variables
   use cudafor

   type :: data_str_def

!=============
! single number quantities
!=============

      integer                       :: i, j 
      real(kind=8)                  :: a 

!=============
! Arrays
!=============

      real(kind=8),   allocatable   :: b(:)
      real(kind=8),   allocatable   :: c(:,:)
      real(kind=8),   allocatable   :: d(:,:,:)
      real(kind=8),   allocatable   :: e(:,:,:,:)

   end type data_str_def

!=============
! Actual data is here
!=============

   type(data_str_def), device, allocatable   :: data_str(:)

   contains

!=============
! subroutine to allocate memory
!=============

      subroutine allocate_mem(n1)
      implicit none 
      integer, intent(in)  :: n1 

      call deallocate_mem()

      write(*,*) 'works here'
      allocate(data_str(n1))

      write(*,*) 'what about allocating memory?'
      allocate(data_str(n1) % b(10))
      write(*,*) 'success!'

      return
      end subroutine allocate_mem

!=============
! subroutine to deallocate memory
!=============

      subroutine deallocate_mem()
      implicit none
      if(allocated(data_str)) deallocate(data_str)
      return 
      end subroutine deallocate_mem

   end module GPU_variables

主程序是'gpu_test.F90'，如下所示：

!=============
! main program 
!=============

    program gpu_test
    use gpu_variables
    implicit none

!=============
! local variables
!=============

    integer             :: i, j, n

!=============
! allocate data
!=============

    n       = 2                 ! number of data structures

    call allocate_mem(n)

!=============
! dallocate device data structures and exit
!=============

    call deallocate_mem()
    end program

编译命令（来自当前文件夹）是：

pgfortran -Mcuda=cc5x *.F90

终端输出：

$ ./a.out 
 works here
 what about allocating memory?
Segmentation fault (core dumped)

任何帮助/见解和解决方案将不胜感激。不，使用指针不是一个可行的选择。

编辑：另一个可能相关的细节：我正在使用 pgfortran 版本 16.10

4

2 回答 2

1

分段错误的原因是您必须访问主机上 data_str 的内存才能分配 data_str(n1)%b。由于 data_str 位于设备内存中，而不是主机内存中，因此您会遇到分段错误。理论上，编译器可以创建一个主机 temp，对其进行分配，然后将其复制到 data_str(n1)%b 的描述符中，但这不是当今 CUDA Fortran 的一部分。

您可以通过自己创建 temp 来解决这种情况：

      subroutine allocate_mem(n1)
      implicit none
      integer, intent(in)  :: n1
      type(data_str_def) :: data_str_h

      call deallocate_mem()

      write(*,*) 'works here'
      allocate(data_str(n1))

      write(*,*) 'what about allocating memory?'
      allocate(data_str_h% b(10))
      data_str(n1) = data_str_h
      write(*,*) 'success!'

      return
      end subroutine allocate_mem

顺便说一句，您打算将组件 b、c、d 和 e 分配在主机内存或设备内存中吗？我没有看到他们的设备属性，所以在上面，他们会去主机内存。

于 2017-07-24T17:31:19.137 回答
1

所以我在 PGI 论坛上发布了这个问题，来自 PGI 的一个人确认该功能不受支持，因为我正在尝试使用它。

http://www.pgroup.com/userforum/viewtopic.php?t=5661

他的建议是使用“托管”属性或在数据结构中使用固定大小的数组。

于 2017-08-10T05:18:09.377 回答