diff --git a/DataFormats/HcalDetId/test/BuildFile.xml b/DataFormats/HcalDetId/test/BuildFile.xml index cbcac315bb5e6..db4e4e2f033c3 100644 --- a/DataFormats/HcalDetId/test/BuildFile.xml +++ b/DataFormats/HcalDetId/test/BuildFile.xml @@ -2,6 +2,6 @@ + - diff --git a/DataFormats/HcalDetId/test/test_hcal_detid.cu b/DataFormats/HcalDetId/test/test_hcal_detid.cu index 3f885984a9778..a10b19adbe0df 100644 --- a/DataFormats/HcalDetId/test/test_hcal_detid.cu +++ b/DataFormats/HcalDetId/test/test_hcal_detid.cu @@ -1,10 +1,12 @@ -#include +#include +#include + #include +#include -#include -#include #include "DataFormats/DetId/interface/DetId.h" #include "DataFormats/HcalDetId/interface/HcalDetId.h" +#include "HeterogeneousCore/CUDAUtilities/interface/requireDevices.h" __global__ void test_gen_detid(DetId *id) { DetId did; @@ -63,17 +65,13 @@ void test_hcal_detid() { } int main(int argc, char **argv) { - int nDevices; - cudaGetDeviceCount(&nDevices); - std::cout << "nDevices = " << nDevices << std::endl; + cms::cudatest::requireDevices(); // test det id functionality - if (nDevices > 0) - test_detid(); + test_detid(); // test hcal det ids - if (nDevices > 0) - test_hcal_detid(); + test_hcal_detid(); return 0; } diff --git a/DataFormats/HcalDigi/test/BuildFile.xml b/DataFormats/HcalDigi/test/BuildFile.xml index c3de08b0c2357..a7d7394a2f951 100644 --- a/DataFormats/HcalDigi/test/BuildFile.xml +++ b/DataFormats/HcalDigi/test/BuildFile.xml @@ -11,6 +11,6 @@ + - diff --git a/DataFormats/HcalDigi/test/test_hcal_digi.cu b/DataFormats/HcalDigi/test/test_hcal_digi.cu index 295607e5abb62..907167ac6ad10 100644 --- a/DataFormats/HcalDigi/test/test_hcal_digi.cu +++ b/DataFormats/HcalDigi/test/test_hcal_digi.cu @@ -1,15 +1,17 @@ -#include +#include +#include + #include +#include -#include -#include +#include "DataFormats/Common/interface/DataFrame.h" #include "DataFormats/DetId/interface/DetId.h" #include "DataFormats/HcalDetId/interface/HcalDetId.h" #include "DataFormats/HcalDigi/interface/HBHEDataFrame.h" +#include "DataFormats/HcalDigi/interface/HcalDigiCollections.h" #include "DataFormats/HcalDigi/interface/QIE10DataFrame.h" #include "DataFormats/HcalDigi/interface/QIE11DataFrame.h" -#include "DataFormats/HcalDigi/interface/HcalDigiCollections.h" -#include "DataFormats/Common/interface/DataFrame.h" +#include "HeterogeneousCore/CUDAUtilities/interface/requireDevices.h" __global__ void kernel_test_hcal_qiesample(HcalQIESample *sample, uint16_t value) { printf("kernel: testing hcal qie sampel\n"); @@ -69,7 +71,6 @@ void test_hcal_qie1011_digis() { constexpr int samples = 10; constexpr int detid = 2; HcalDataFrameContainer coll{samples, detid}; - TDF *d_dfs; uint16_t *d_data; uint32_t *d_out; uint32_t h_out[size], h_test_out[size]; @@ -162,22 +163,18 @@ void test_hcal_qie8_hbhedf() { } int main(int argc, char **argv) { - int nDevices; - cudaGetDeviceCount(&nDevices); - std::cout << "nDevices = " << nDevices << std::endl; - - if (nDevices > 0) { - // qie8 - test_hcal_qiesample(); - test_hcal_qie8_hbhedf(); - test_hcal_qie8_digis(); - test_hcal_qie8_digis(); - test_hcal_qie8_digis(); - - // qie1011 - test_hcal_qie1011_digis(); - test_hcal_qie1011_digis(); - } + cms::cudatest::requireDevices(); + + // qie8 + test_hcal_qiesample(); + test_hcal_qie8_hbhedf(); + test_hcal_qie8_digis(); + test_hcal_qie8_digis(); + test_hcal_qie8_digis(); + + // qie1011 + test_hcal_qie1011_digis(); + test_hcal_qie1011_digis(); return 0; } diff --git a/DataFormats/HcalRecHit/test/BuildFile.xml b/DataFormats/HcalRecHit/test/BuildFile.xml index 0b9deb8f1e8f3..a28e7783fe290 100644 --- a/DataFormats/HcalRecHit/test/BuildFile.xml +++ b/DataFormats/HcalRecHit/test/BuildFile.xml @@ -7,11 +7,11 @@ + - + - diff --git a/DataFormats/HcalRecHit/test/test_hcal_reco.cu b/DataFormats/HcalRecHit/test/test_hcal_reco.cu index 4af8c3c065e2c..5f5d39fc562a5 100644 --- a/DataFormats/HcalRecHit/test/test_hcal_reco.cu +++ b/DataFormats/HcalRecHit/test/test_hcal_reco.cu @@ -10,6 +10,7 @@ #include "DataFormats/HcalRecHit/interface/HORecHit.h" #include "DataFormats/HcalRecHit/interface/HFQIE10Info.h" #include "DataFormats/HcalRecHit/interface/HBHEChannelInfo.h" +#include "HeterogeneousCore/CUDAUtilities/interface/requireDevices.h" template __global__ void kernel_test_hcal_rechits(T *other) { @@ -109,18 +110,13 @@ void test_hcal_hbhechinfo() { } int main(int argc, char **argv) { - int nDevices; - cudaGetDeviceCount(&nDevices); - std::cout << "nDevices = " << nDevices << std::endl; + cms::cudatest::requireDevices(); - if (nDevices > 0) { - test_hcal_rechits(); - test_hcal_rechits(); - test_hcal_rechits(); - test_hcal_hbhechinfo(); - - std::cout << "all good" << std::endl; - } + test_hcal_rechits(); + test_hcal_rechits(); + test_hcal_rechits(); + test_hcal_hbhechinfo(); + std::cout << "all good" << std::endl; return 0; }