File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
mlir/lib/Dialect/LLVMIR/IR Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change 12
12
// ===----------------------------------------------------------------------===//
13
13
14
14
#include " mlir/Dialect/LLVMIR/BasicPtxBuilderInterface.h"
15
- #include " mlir/Dialect/LLVMIR/NVVMDialect.h"
16
15
17
16
#define DEBUG_TYPE " ptx-builder"
18
17
#define DBGS () (llvm::dbgs() << " [" DEBUG_TYPE " ]: " )
27
26
using namespace mlir ;
28
27
using namespace NVVM ;
29
28
29
+ static constexpr int64_t kSharedMemorySpace = 3 ;
30
+
30
31
static char getRegisterType (Type type) {
31
32
if (type.isInteger (1 ))
32
33
return ' b' ;
@@ -42,7 +43,7 @@ static char getRegisterType(Type type) {
42
43
return ' d' ;
43
44
if (auto ptr = dyn_cast<LLVM::LLVMPointerType>(type)) {
44
45
// Shared address spaces is addressed with 32-bit pointers.
45
- if (ptr.getAddressSpace () == NVVMMemorySpace:: kSharedMemorySpace ) {
46
+ if (ptr.getAddressSpace () == kSharedMemorySpace ) {
46
47
return ' r' ;
47
48
}
48
49
return ' l' ;
You can’t perform that action at this time.
0 commit comments