1 //===-- NVPTXutil.cpp - Functions exported to CodeGen --*- C++ -*-===// 2 // 3 // The LLVM Compiler Infrastructure 4 // 5 // This file is distributed under the University of Illinois Open Source 6 // License. See LICENSE.TXT for details. 7 // 8 //===----------------------------------------------------------------------===// 9 // 10 // This file contains the functions that can be used in CodeGen. 11 // 12 //===----------------------------------------------------------------------===// 13 14 #include "NVPTXutil.h" 15 #include "NVPTX.h" 16 17 using namespace llvm; 18 19 namespace llvm { 20 21 bool isParamLoad(const MachineInstr *MI) 22 { 23 if ((MI->getOpcode() != NVPTX::LD_i32_avar) && 24 (MI->getOpcode() != NVPTX::LD_i64_avar)) 25 return false; 26 if (MI->getOperand(2).isImm() == false) 27 return false; 28 if (MI->getOperand(2).getImm() != NVPTX::PTXLdStInstCode::PARAM) 29 return false; 30 return true; 31 } 32 33 #define DATA_MASK 0x7f 34 #define DIGIT_WIDTH 7 35 #define MORE_BYTES 0x80 36 37 static int encode_leb128(uint64_t val, int *nbytes, 38 char *space, int splen) 39 { 40 char *a; 41 char *end = space + splen; 42 43 a = space; 44 do { 45 unsigned char uc; 46 47 if (a >= end) 48 return 1; 49 uc = val & DATA_MASK; 50 val >>= DIGIT_WIDTH; 51 if (val != 0) 52 uc |= MORE_BYTES; 53 *a = uc; 54 a++; 55 } while (val); 56 *nbytes = a - space; 57 return 0; 58 } 59 60 #undef DATA_MASK 61 #undef DIGIT_WIDTH 62 #undef MORE_BYTES 63 64 uint64_t encode_leb128(const char *str) 65 { 66 union { uint64_t x; char a[8]; } temp64; 67 68 temp64.x = 0; 69 70 for (unsigned i=0,e=strlen(str); i!=e; ++i) 71 temp64.a[i] = str[e-1-i]; 72 73 char encoded[16]; 74 int nbytes; 75 76 int retval = encode_leb128(temp64.x, &nbytes, encoded, 16); 77 78 (void)retval; 79 assert(retval == 0 && 80 "Encoding to leb128 failed"); 81 82 assert(nbytes <= 8 && 83 "Cannot support register names with leb128 encoding > 8 bytes"); 84 85 temp64.x = 0; 86 for (int i=0; i<nbytes; ++i) 87 temp64.a[i] = encoded[i]; 88 89 return temp64.x; 90 } 91 92 } // end namespace llvm 93