File tree 1 file changed +3
-2
lines changed
mlir/lib/Dialect/LLVMIR/IR
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
#include " mlir/Support/LogicalResult.h"
17
16
18
17
#define DEBUG_TYPE " ptx-builder"
28
27
using namespace mlir ;
29
28
using namespace NVVM ;
30
29
30
+ static constexpr int64_t kSharedMemorySpace = 3 ;
31
+
31
32
static char getRegisterType (Type type) {
32
33
if (type.isInteger (1 ))
33
34
return ' b' ;
@@ -43,7 +44,7 @@ static char getRegisterType(Type type) {
43
44
return ' d' ;
44
45
if (auto ptr = type.dyn_cast <LLVM::LLVMPointerType>()) {
45
46
// Shared address spaces is addressed with 32-bit pointers.
46
- if (ptr.getAddressSpace () == NVVM:: kSharedMemorySpace ) {
47
+ if (ptr.getAddressSpace () == kSharedMemorySpace ) {
47
48
return ' r' ;
48
49
}
49
50
return ' l' ;
You can’t perform that action at this time.
0 commit comments