derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
262 lines
13 KiB
Fortran
262 lines
13 KiB
Fortran
!! Copyright (c) 2017-2023 Advanced Micro Devices, Inc. All rights reserved.
|
|
!!
|
|
!! Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
!! of this software and associated documentation files (the "Software"), to deal
|
|
!! in the Software without restriction, including without limitation the rights
|
|
!! to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
!! copies of the Software, and to permit persons to whom the Software is
|
|
!! furnished to do so, subject to the following conditions:
|
|
!!
|
|
!! The above copyright notice and this permission notice shall be included in
|
|
!! all copies or substantial portions of the Software.
|
|
!!
|
|
!! THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
!! IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
!! FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
!! AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
!! LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
!! OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
!! THE SOFTWARE.
|
|
|
|
module test_rocrand
|
|
use fruit
|
|
use fruit_helpers
|
|
use hipfor
|
|
use rocrand_m
|
|
|
|
implicit none
|
|
|
|
contains
|
|
|
|
!> Initialise test suite - no-op.
|
|
subroutine setup
|
|
end subroutine setup
|
|
|
|
!> Clean up test suite - no-op.
|
|
subroutine teardown
|
|
end subroutine teardown
|
|
|
|
!> Test generator functions
|
|
subroutine test_init()
|
|
integer(kind =8) :: gen
|
|
integer(kind =8), parameter :: seed = 5, offset = 10
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_create_generator(gen, &
|
|
ROCRAND_RNG_PSEUDO_DEFAULT))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_initialize_generator(gen))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_set_seed(gen, seed))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_set_offset(gen, offset))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_destroy_generator(gen))
|
|
end subroutine test_init
|
|
|
|
!> Test rocrand_generate.
|
|
subroutine test_rocrand_generate()
|
|
integer(kind =8) :: gen
|
|
integer(kind =4), target, dimension(128) :: h_x
|
|
type(c_ptr) :: d_x
|
|
integer(c_size_t), parameter :: output_size = 128
|
|
call assert_equals(hipSuccess, hipMalloc(d_x, output_size * sizeof(h_x(1))))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_create_generator(gen, &
|
|
ROCRAND_RNG_PSEUDO_DEFAULT))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_generate(gen, d_x, output_size))
|
|
call assert_equals(hipSuccess, hipMemcpy(c_loc(h_x), d_x, output_size * sizeof(h_x(1)), &
|
|
hipMemcpyDeviceToHost))
|
|
call assert_equals(hipSuccess, hipFree(d_x))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_destroy_generator(gen))
|
|
end subroutine test_rocrand_generate
|
|
|
|
!> Test rocrand_generate using a host generator.
|
|
subroutine test_rocrand_generate_host()
|
|
integer(kind =8) :: gen
|
|
integer(kind =4), target, dimension(128) :: h_x
|
|
integer(c_size_t), parameter :: output_size = 128
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_create_generator_host(gen, &
|
|
ROCRAND_RNG_PSEUDO_MRG31K3P))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_generate(gen, c_loc(h_x), output_size))
|
|
call assert_equals(hipSuccess, hipDeviceSynchronize())
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_destroy_generator(gen))
|
|
end subroutine test_rocrand_generate_host
|
|
|
|
!> Test rocrand_generate_uniform.
|
|
subroutine test_rocrand_generate_uniform()
|
|
integer(kind =8) :: gen
|
|
real, target, dimension(128) :: h_x
|
|
type(c_ptr) :: d_x
|
|
integer(c_size_t), parameter :: output_size = 128
|
|
real, parameter :: mean = 0.5, delta = 0.1
|
|
call assert_equals(hipSuccess, hipMalloc(d_x, output_size * sizeof(h_x(1))))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_create_generator(gen, &
|
|
ROCRAND_RNG_PSEUDO_DEFAULT))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_generate_uniform(gen, d_x, output_size))
|
|
call assert_equals(hipSuccess, hipMemcpy(c_loc(h_x), d_x, output_size * sizeof(h_x(1)), &
|
|
hipMemcpyDeviceToHost))
|
|
call assert_equals((sum(h_x) / output_size), mean, delta)
|
|
call assert_equals(hipSuccess, hipFree(d_x))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_destroy_generator(gen))
|
|
end subroutine test_rocrand_generate_uniform
|
|
|
|
!> Test rocrand_generate_uniform_double.
|
|
subroutine test_rocrand_generate_uniform_double()
|
|
integer(kind =8) :: gen
|
|
double precision, target, dimension(128) :: h_x
|
|
type(c_ptr) :: d_x
|
|
integer(c_size_t), parameter :: output_size = 128
|
|
double precision, parameter :: mean = 0.5, delta = 0.1
|
|
call assert_equals(hipSuccess, hipMalloc(d_x, output_size * sizeof(h_x(1))))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_create_generator(gen, &
|
|
ROCRAND_RNG_PSEUDO_DEFAULT))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_generate_uniform_double(gen, d_x, &
|
|
output_size))
|
|
call assert_equals(hipSuccess, hipMemcpy(c_loc(h_x), d_x, output_size * sizeof(h_x(1)), &
|
|
hipMemcpyDeviceToHost))
|
|
call assert_equals((sum(h_x) / output_size), mean, delta)
|
|
call assert_equals(hipSuccess, hipFree(d_x))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_destroy_generator(gen))
|
|
end subroutine test_rocrand_generate_uniform_double
|
|
|
|
!> Test rocrand_generate_normal.
|
|
subroutine test_rocrand_generate_normal()
|
|
integer(kind =8) :: gen
|
|
real, target, dimension(128) :: h_x
|
|
type(c_ptr) :: d_x
|
|
integer(c_size_t), parameter :: output_size = 128
|
|
real, parameter :: mean = 0.0, stddev = 1.0, delta = 0.2
|
|
call assert_equals(hipSuccess, hipMalloc(d_x, output_size * sizeof(h_x(1))))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_create_generator(gen, &
|
|
ROCRAND_RNG_PSEUDO_DEFAULT))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_generate_normal(gen, d_x, &
|
|
output_size, mean, stddev))
|
|
call assert_equals(hipSuccess, hipMemcpy(c_loc(h_x), d_x, output_size * sizeof(h_x(1)), &
|
|
hipMemcpyDeviceToHost))
|
|
call assert_equals((sum(h_x) / output_size), mean, delta)
|
|
call assert_equals(sqrt(sum((h_x - mean) ** 2) / output_size), stddev, delta)
|
|
call assert_equals(hipSuccess, hipFree(d_x))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_destroy_generator(gen))
|
|
end subroutine test_rocrand_generate_normal
|
|
|
|
!> Test rocrand_generate_normal_double.
|
|
subroutine test_rocrand_generate_normal_double()
|
|
integer(kind =8) :: gen
|
|
double precision, target, dimension(128) :: h_x
|
|
type(c_ptr) :: d_x
|
|
integer(c_size_t), parameter :: output_size = 128
|
|
double precision, parameter :: mean = 0.0, stddev = 1.0, delta = 0.2
|
|
call assert_equals(hipSuccess, hipMalloc(d_x, output_size * sizeof(h_x(1))))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_create_generator(gen, &
|
|
ROCRAND_RNG_PSEUDO_DEFAULT))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_generate_normal_double(gen, d_x, &
|
|
output_size, mean, stddev))
|
|
call assert_equals(hipSuccess, hipMemcpy(c_loc(h_x), d_x, output_size * sizeof(h_x(1)), &
|
|
hipMemcpyDeviceToHost))
|
|
call assert_equals((sum(h_x) / output_size), mean, delta)
|
|
call assert_equals(sqrt(sum((h_x - mean) ** 2) / output_size), stddev, delta)
|
|
call assert_equals(hipSuccess, hipFree(d_x))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_destroy_generator(gen))
|
|
end subroutine test_rocrand_generate_normal_double
|
|
|
|
!> Test rocrand_generate_log_normal.
|
|
subroutine test_rocrand_generate_log_normal()
|
|
integer(kind =8) :: gen
|
|
real, target, dimension(128) :: h_x
|
|
type(c_ptr) :: d_x
|
|
integer(c_size_t), parameter :: output_size = 128
|
|
real, parameter :: mean = 1.6, stddev = 0.25, delta = 0.2
|
|
real :: m, s
|
|
call assert_equals(hipSuccess, hipMalloc(d_x, output_size * sizeof(h_x(1))))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_create_generator(gen, &
|
|
ROCRAND_RNG_PSEUDO_DEFAULT))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_generate_log_normal(gen, d_x, &
|
|
output_size, mean, stddev))
|
|
call assert_equals(hipSuccess, hipMemcpy(c_loc(h_x), d_x, output_size * sizeof(h_x(1)), &
|
|
hipMemcpyDeviceToHost))
|
|
m = sum(h_x) / output_size
|
|
s = sqrt(sum((h_x - mean) ** 2) / output_size)
|
|
call assert_equals(log(m * m / sqrt(s + m * m)), mean, delta)
|
|
call assert_equals(sqrt(log(1.0 + s / (m * m))), stddev, delta)
|
|
call assert_equals(hipSuccess, hipFree(d_x))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_destroy_generator(gen))
|
|
end subroutine test_rocrand_generate_log_normal
|
|
|
|
!> Test rocrand_generate_log_normal_double.
|
|
subroutine test_rocrand_generate_log_normal_double()
|
|
integer(kind =8) :: gen
|
|
double precision, target, dimension(128) :: h_x
|
|
type(c_ptr) :: d_x
|
|
integer(c_size_t), parameter :: output_size = 128
|
|
double precision, parameter :: mean = 1.6, stddev = 0.25, delta = 0.2
|
|
double precision :: m, s
|
|
call assert_equals(hipSuccess, hipMalloc(d_x, output_size * sizeof(h_x(1))))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_create_generator(gen, &
|
|
ROCRAND_RNG_PSEUDO_DEFAULT))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_generate_log_normal_double(gen, d_x, &
|
|
output_size, mean, stddev))
|
|
call assert_equals(hipSuccess, hipMemcpy(c_loc(h_x), d_x, output_size * sizeof(h_x(1)), &
|
|
hipMemcpyDeviceToHost))
|
|
m = sum(h_x) / output_size
|
|
s = sqrt(sum((h_x - mean) ** 2) / output_size)
|
|
call assert_equals(log(m * m / sqrt(s + m * m)), mean, delta)
|
|
call assert_equals(sqrt(log(1.0 + s / (m * m))), stddev, delta)
|
|
call assert_equals(hipSuccess, hipFree(d_x))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_destroy_generator(gen))
|
|
end subroutine test_rocrand_generate_log_normal_double
|
|
|
|
!> Test rocrand_generate_poisson.
|
|
subroutine test_rocrand_generate_poisson()
|
|
integer(kind =8) :: gen, distribution
|
|
integer(kind =4), target, dimension(128) :: h_x
|
|
type(c_ptr) :: d_x
|
|
integer(c_size_t), parameter :: output_size = 128
|
|
double precision, parameter :: lambda = 20.0
|
|
call assert_equals(hipSuccess, hipMalloc(d_x, output_size * sizeof(h_x(1))))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_create_generator(gen, &
|
|
ROCRAND_RNG_PSEUDO_DEFAULT))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_create_poisson_distribution(lambda, &
|
|
distribution))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_generate_poisson(gen, d_x, &
|
|
output_size, lambda))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_destroy_discrete_distribution( &
|
|
distribution))
|
|
call assert_equals(hipSuccess, hipMemcpy(c_loc(h_x), d_x, output_size * sizeof(h_x(1)), &
|
|
hipMemcpyDeviceToHost))
|
|
call assert_equals(hipSuccess, hipFree(d_x))
|
|
call assert_equals(ROCRAND_STATUS_SUCCESS, rocrand_destroy_generator(gen))
|
|
end subroutine test_rocrand_generate_poisson
|
|
|
|
!> Call each test.
|
|
subroutine rocrand_basket()
|
|
character(len=*) :: suite_name
|
|
parameter(suite_name='test_rocrand')
|
|
|
|
call run_fruit_test_case(test_init,'test_init',&
|
|
setup,teardown,suite_name)
|
|
|
|
call run_fruit_test_case(test_rocrand_generate,'test_rocrand_generate',&
|
|
setup,teardown,suite_name)
|
|
|
|
call run_fruit_test_case(test_rocrand_generate_host,'test_rocrand_generate_host',&
|
|
setup,teardown,suite_name)
|
|
|
|
call run_fruit_test_case(test_rocrand_generate_uniform,'test_rocrand_generate_uniform',&
|
|
setup,teardown,suite_name)
|
|
|
|
call run_fruit_test_case(test_rocrand_generate_uniform_double, &
|
|
'test_rocrand_generate_uniform_double',setup,teardown,suite_name)
|
|
|
|
call run_fruit_test_case(test_rocrand_generate_normal,'test_rocrand_generate_normal',&
|
|
setup,teardown,suite_name)
|
|
|
|
call run_fruit_test_case(test_rocrand_generate_normal_double, &
|
|
'test_rocrand_generate_normal_double',setup,teardown,suite_name)
|
|
|
|
call run_fruit_test_case(test_rocrand_generate_log_normal,'test_rocrand_generate_log_normal',&
|
|
setup,teardown,suite_name)
|
|
|
|
call run_fruit_test_case(test_rocrand_generate_log_normal_double, &
|
|
'test_rocrand_generate_log_normal_double',setup,teardown,suite_name)
|
|
|
|
call run_fruit_test_case(test_rocrand_generate_poisson,'test_rocrand_generate_poisson',&
|
|
setup,teardown,suite_name)
|
|
|
|
end subroutine rocrand_basket
|
|
|
|
end module test_rocrand
|