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 17using namespace llvm; 18 19namespace llvm { 20 21bool isParamLoad(const MachineInstr *MI) { 22 if ((MI->getOpcode() != NVPTX::LD_i32_avar) && 23 (MI->getOpcode() != NVPTX::LD_i64_avar)) 24 return false; 25 if (MI->getOperand(2).isImm() == false) 26 return false; 27 if (MI->getOperand(2).getImm() != NVPTX::PTXLdStInstCode::PARAM) 28 return false; 29 return true; 30} 31 32#define DATA_MASK 0x7f 33#define DIGIT_WIDTH 7 34#define MORE_BYTES 0x80 35 36static int encode_leb128(uint64_t val, int *nbytes, char *space, int splen) { 37 char *a; 38 char *end = space + splen; 39 40 a = space; 41 do { 42 unsigned char uc; 43 44 if (a >= end) 45 return 1; 46 uc = val & DATA_MASK; 47 val >>= DIGIT_WIDTH; 48 if (val != 0) 49 uc |= MORE_BYTES; 50 *a = uc; 51 a++; 52 } while (val); 53 *nbytes = a - space; 54 return 0; 55} 56 57#undef DATA_MASK 58#undef DIGIT_WIDTH 59#undef MORE_BYTES 60 61uint64_t encode_leb128(const char *str) { 62 union { 63 uint64_t x; 64 char a[8]; 65 } temp64; 66 67 temp64.x = 0; 68 69 for (unsigned i = 0, e = strlen(str); i != e; ++i) 70 temp64.a[i] = str[e - 1 - i]; 71 72 char encoded[16]; 73 int nbytes; 74 75 int retval = encode_leb128(temp64.x, &nbytes, encoded, 16); 76 77 (void) retval; 78 assert(retval == 0 && "Encoding to leb128 failed"); 79 80 assert(nbytes <= 8 && 81 "Cannot support register names with leb128 encoding > 8 bytes"); 82 83 temp64.x = 0; 84 for (int i = 0; i < nbytes; ++i) 85 temp64.a[i] = encoded[i]; 86 87 return temp64.x; 88} 89 90} // end namespace llvm 91