youngkingdom/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Replaced GoogleTest copy with submodule. Added updates to support intra-threadblock reductions. Added tests for same.

Browse Source
This commit is contained in:
akerr
2018-06-11 11:47:15 -07:00
parent 2c496c3e9e
commit 374882be53
40 changed files with 3279 additions and 336 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tools/test/unit/cutlass_unit_test.cpp
View File

@ -26,9 +26,26 @@
\brief CUTLASS Unit Tests
*/
#include <cuda_runtime_api.h>
#include <gtest/gtest.h>
void set_gtest_flag() {
// Default flags can be overwritten by --gtest_filter from commandline
cudaDeviceProp deviceProperties;
cudaGetDeviceProperties(&deviceProperties, 0);
int deviceMajorMinor = deviceProperties.major * 10 + deviceProperties.minor;
if (deviceMajorMinor < 53)
::testing::GTEST_FLAG(filter) = "-*Igemm*:*Hgemm*:*mma*";
else if (deviceMajorMinor < 61)
::testing::GTEST_FLAG(filter) = "-*Igemm*:*mma*";
else if (deviceMajorMinor < 70)
::testing::GTEST_FLAG(filter) = "-*mma*";
}
int main(int argc, char* arg[]) {
set_gtest_flag();
::testing::InitGoogleTest(&argc, arg);
return RUN_ALL_TESTS();
}