hlrs-software-stack/spack
7
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add smoke test to VTK-m package (#19816)

Browse source 
* add smoke test

* remove whitespaces

* fix minimum version issue

* reorder decorators & replace make with cmake build

* merge cmake build into one line

* reorganize smoke test function

Co-authored-by: Jieyang Chen <chenj3@ornl.gov>
This commit is contained in:
Jieyang Chen 2020-12-01 18:59:44 -05:00 committed by GitHub
parent 59106c4dec
commit c0bbf5454c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 169 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

169
var/spack/repos/builtin/packages/vtk-m/package.py
View file

@ -6,6 +6,7 @@
from spack import *
import os
import shutil
import sys
@ -187,3 +188,171 @@ def cmake_args(self):
options.append("-DVTKm_ENABLE_TBB:BOOL=OFF")
return options
def smoke_test(self):
print("Checking VTK-m installation...")
spec = self.spec
checkdir = "spack-check"
with working_dir(checkdir, create=True):
source = r"""
#include <vtkm/cont/Algorithm.h>
#include <vtkm/cont/ArrayHandle.h>
#include <vtkm/cont/Initialize.h>
#include <iostream>
#include <vector>
struct NoArgKernel {
VTKM_EXEC void operator()(vtkm::Id) const {}
void SetErrorMessageBuffer(
const vtkm::exec::internal::ErrorMessageBuffer &errorMessage) {
this->ErrorMessage = errorMessage;
}
vtkm::exec::internal::ErrorMessageBuffer ErrorMessage;
};
template <typename PortalType> struct FillArrayKernel {
using ValueType = typename PortalType::ValueType;
FillArrayKernel(const PortalType &array, ValueType fill)
: Array(array), FillValue(fill) {}
VTKM_EXEC void operator()(vtkm::Id index) const {
this->Array.Set(index, this->FillValue);
}
void SetErrorMessageBuffer(
const vtkm::exec::internal::ErrorMessageBuffer &) {
}
PortalType Array;
ValueType FillValue;
};
int main() {
vtkm::cont::Initialize();
constexpr vtkm::Id size = 1000000;
#if defined(VTKM_ENABLE_KOKKOS)
constexpr vtkm::cont::DeviceAdapterTagKokkos desired_device;
#elif defined(VTKM_ENABLE_CUDA)
constexpr vtkm::cont::DeviceAdapterTagCuda desired_device;
#elif defined(VTKM_ENABLE_TBB)
constexpr vtkm::cont::DeviceAdapterTagTBB desired_device;
#elif defined(VTKM_ENABLE_OPENMP)
constexpr vtkm::cont::DeviceAdapterTagOpenMP desired_device;
#else
#error "No VTK-m Device Adapter enabled"
#endif
std::cout << "-------------------------------------------\n";
std::cout << "Testing No Argument Kernel" << std::endl;
vtkm::cont::Algorithm::Schedule(desired_device, NoArgKernel(), size);
vtkm::cont::ArrayHandle<vtkm::Id> handle;
{
std::cout << "-------------------------------------------\n";
std::cout << "Testing Kernel + ArrayHandle PrepareForOutput" << std::endl;
vtkm::cont::Token token;
auto portal = handle.PrepareForOutput(size, desired_device, token);
vtkm::cont::Algorithm::Schedule(desired_device,
FillArrayKernel<decltype(portal)>{portal, 1}, size);
}
{
std::cout << "-------------------------------------------\n";
std::cout << "Testing Kernel + ArrayHandle PrepareForInPlace" << std::endl;
vtkm::cont::Token token;
auto portal = handle.PrepareForInPlace(desired_device, token);
vtkm::cont::Algorithm::Schedule(desired_device,
FillArrayKernel<decltype(portal)>{portal, 2}, size);
}
std::cout << "-------------------------------------------\n";
std::cout << "Ran tests on: " << desired_device.GetName() << std::endl;
return 0;
}
"""
cmakelists = r"""
##============================================================================
## Copyright (c) Kitware, Inc.
## All rights reserved.
## See LICENSE.txt for details.
##
## This software is distributed WITHOUT ANY WARRANTY; without even
## the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
## PURPOSE. See the above copyright notice for more information.
##============================================================================
cmake_minimum_required(VERSION 3.12...3.18 FATAL_ERROR)
project(VTKmSmokeTest CXX)
#Find the VTK-m package
find_package(VTKm REQUIRED QUIET)
add_executable(VTKmSmokeTest main.cxx)
target_link_libraries(VTKmSmokeTest PRIVATE vtkm_cont)
vtkm_add_target_information(VTKmSmokeTest
DROP_UNUSED_SYMBOLS MODIFY_CUDA_FLAGS
DEVICE_SOURCES main.cxx)
"""
with open("main.cxx", 'w') as f:
f.write(source)
with open("CMakeLists.txt", 'w') as f:
f.write(cmakelists)
builddir = "build"
with working_dir(builddir, create=True):
cmake = Executable(spec['cmake'].prefix.bin + "/cmake")
cmakefiledir = spec['vtk-m'].prefix.lib + "/cmake"
cmakefiledir = cmakefiledir + "/" + os.listdir(cmakefiledir)[0]
cmake(*(["..", "-DVTKm_DIR=" + cmakefiledir]))
cmake(*(["--build", "."]))
try:
test = Executable('./VTKmSmokeTest')
output = test(output=str)
except ProcessError:
output = ""
print(output)
if "+hip" in spec:
expected_device = 'Kokkos'
elif "+cuda" in spec:
expected_device = 'Cuda'
elif "+tbb" in spec:
expected_device = 'TBB'
elif "+openmp" in spec:
expected_device = 'OpenMP'
expected = """\
-------------------------------------------
Testing No Argument Kernel
-------------------------------------------
Testing Kernel + ArrayHandle PrepareForOutput
-------------------------------------------
Testing Kernel + ArrayHandle PrepareForInPlace
-------------------------------------------
Ran tests on: """ + expected_device + "\n"
success = output == expected
if success:
print("Test success")
if not success:
print("Produced output does not match expected output.")
print("Expected output:")
print('-' * 80)
print(expected)
print('-' * 80)
print("Produced output:")
print('-' * 80)
print(output)
print('-' * 80)
raise RuntimeError("VTK-m install check failed")
shutil.rmtree(checkdir)
@run_after('install')
@on_package_attributes(run_tests=True)
def check_install(self):
spec = self.spec
if "@master" in spec:
self.smoke_test()