CUTLASS 3.0.0 (#786)

* CUTLASS 3.0.0
This commit is contained in:
Vijay Thakkar
2023-01-23 20:55:28 -05:00
committed by GitHub
parent 66d9cddc83
commit 277bd6e537
377 changed files with 76396 additions and 1186 deletions
+23 -1
View File
@@ -39,6 +39,17 @@
#include <cstdlib>
#include <string>
#include <cuda_runtime_api.h>
/////////////////////////////////////////////////////////////////////////////////////////////////
/// Gets a CUDA device
cudaDeviceProp GetCudaDevice();
/// Prints device properties
std::ostream &operator<<(std::ostream &out, cudaDeviceProp const &device);
/////////////////////////////////////////////////////////////////////////////////////////////////
/// Sets flags for Unit test
@@ -52,7 +63,6 @@ int CutlassUnitTestProblemCount();
/////////////////////////////////////////////////////////////////////////////////////////////////
// active test macro
#define CUTLASS_TEST_LEVEL_ACTIVE(LEVEL,NAME_STATIC,NAME_DYNAMIC,...) \
TEST(NAME_STATIC,L##LEVEL##_##NAME_DYNAMIC) __VA_ARGS__
@@ -78,3 +88,15 @@ int CutlassUnitTestProblemCount();
#if !defined(CUTLASS_TEST_UNIT_ENABLE_WARNINGS)
#define CUTLASS_TEST_UNIT_ENABLE_WARNINGS false
#endif
#if (__CUDACC_VER_MAJOR__ >= 12)
#define CUDA_12_0_SM90_FEATURES_SUPPORTED true
#else
#define CUDA_12_0_SM90_FEATURES_SUPPORTED false
#endif
#include <cutlass/cutlass.h>
#include <cutlass/numeric_types.h>
#include <cutlass/trace.h>
/////////////////////////////////////////////////////////////////////////////////////////////////