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 1212// ===----------------------------------------------------------------------===//
1313
1414#include " mlir/Dialect/LLVMIR/BasicPtxBuilderInterface.h"
15- #include " mlir/Dialect/LLVMIR/NVVMDialect.h"
1615#include " mlir/Support/LogicalResult.h"
1716
1817#define DEBUG_TYPE " ptx-builder"
2827using namespace mlir ;
2928using namespace NVVM ;
3029
30+ static constexpr int64_t kSharedMemorySpace = 3 ;
31+
3132static char getRegisterType (Type type) {
3233 if (type.isInteger (1 ))
3334 return ' b' ;
@@ -43,7 +44,7 @@ static char getRegisterType(Type type) {
4344 return ' d' ;
4445 if (auto ptr = type.dyn_cast <LLVM::LLVMPointerType>()) {
4546 // Shared address spaces is addressed with 32-bit pointers.
46- if (ptr.getAddressSpace () == NVVM:: kSharedMemorySpace ) {
47+ if (ptr.getAddressSpace () == kSharedMemorySpace ) {
4748 return ' r' ;
4849 }
4950 return ' l' ;
You can’t perform that action at this time.
0 commit comments