Cutlass
CUDA Templates for Linear Algebra Subroutines and Solvers
cutlass
cutlass.h
Go to the documentation of this file.
1
/***************************************************************************************************
2
* Copyright (c) 2017-2018, NVIDIA CORPORATION. All rights reserved.
3
*
4
* Redistribution and use in source and binary forms, with or without modification, are permitted
5
* provided that the following conditions are met:
6
* * Redistributions of source code must retain the above copyright notice, this list of
7
* conditions and the following disclaimer.
8
* * Redistributions in binary form must reproduce the above copyright notice, this list of
9
* conditions and the following disclaimer in the documentation and/or other materials
10
* provided with the distribution.
11
* * Neither the name of the NVIDIA CORPORATION nor the names of its contributors may be used
12
* to endorse or promote products derived from this software without specific prior written
13
* permission.
14
*
15
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND ANY EXPRESS OR
16
* IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND
17
* FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE
18
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
19
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
20
* OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
21
* STRICT LIABILITY, OR TOR (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
22
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23
*
24
**************************************************************************************************/
25
30
#pragma once
31
33
34
#define CUTLASS_MAJOR 1
35
#define CUTLASS_MINOR 2
36
#define CUTLASS_PATCH 0
37
#define CUTLASS_VERSION ((CUTLASS_MAJOR)*100 + (CUTLASS_MINOR)*10 + CUTLASS_PATCH)
38
39
#ifdef __NVCC__
40
#define CUTLASS_HOST_DEVICE __forceinline__ __device__ __host__
41
#define CUTLASS_DEVICE __forceinline__ __device__
42
#elif defined(__CUDACC_RTC__)
43
#define CUTLASS_HOST_DEVICE __forceinline__ __device__
44
#define CUTLASS_DEVICE __forceinline__ __device__
45
#else
46
#define CUTLASS_HOST_DEVICE
47
// CUTLASS_DEVICE is an error if not compiling device code
48
#endif
49
50
#define CUTLASS_ASSERT(x) assert(x)
51
52
#include "
cutlass/util/performance_tuning.h
"
53
54
// A small helper class to dump a type at compile time
55
// Usage:: DumpType<Class>::Class
56
template
<
typename
T>
57
struct
DebugType
{};
58
59
template
<
typename
T>
60
void
DebugTypeFunc
(T
const
& t) {
61
T::t;
62
}
63
64
// A small helper class to dump a compile time constant at compile time
65
// Usage: DumpValue<Class::kConstant>::kConstant
66
template
<
int
Value>
67
struct
DebugValue
{};
68
69
namespace
cutlass
{
70
72
static
const
int
kWarpSize = 32;
73
74
}
// namespace cutlass
75
cutlass
Definition:
convert.h:33
DebugValue
Definition:
cutlass.h:67
performance_tuning.h
DebugType
Definition:
cutlass.h:57
DebugTypeFunc
void DebugTypeFunc(T const &t)
Definition:
cutlass.h:60
Generated on Fri Oct 26 2018 14:53:33 for Cutlass by
1.8.14