X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=lib%2FTarget%2FPTX%2FPTXISelLowering.cpp;h=3307d91a618803749a7e0a69dcced775d1e2aee3;hb=10820d9a9754987533fd7a8b50b03831781cbd65;hp=a624b5a201b953c4e4697809387b3b9ef07c7366;hpb=37f80658b3237d90b0e6d9fb0b59c214b120898a;p=oota-llvm.git

diff --git a/lib/Target/PTX/PTXISelLowering.cpp b/lib/Target/PTX/PTXISelLowering.cpp
index a624b5a201b..3307d91a618 100644
--- a/lib/Target/PTX/PTXISelLowering.cpp
+++ b/lib/Target/PTX/PTXISelLowering.cpp
@@ -236,23 +236,7 @@ SDValue PTXTargetLowering::
   else {
     for (unsigned i = 0, e = Ins.size(); i != e; ++i) {
       EVT                  RegVT = Ins[i].VT;
-      TargetRegisterClass* TRC   = 0;
-
-      // Determine which register class we need
-      if (RegVT == MVT::i1)
-        TRC = PTX::RegPredRegisterClass;
-      else if (RegVT == MVT::i16)
-        TRC = PTX::RegI16RegisterClass;
-      else if (RegVT == MVT::i32)
-        TRC = PTX::RegI32RegisterClass;
-      else if (RegVT == MVT::i64)
-        TRC = PTX::RegI64RegisterClass;
-      else if (RegVT == MVT::f32)
-        TRC = PTX::RegF32RegisterClass;
-      else if (RegVT == MVT::f64)
-        TRC = PTX::RegF64RegisterClass;
-      else
-        llvm_unreachable("Unknown parameter type");
+      TargetRegisterClass* TRC   = getRegClassFor(RegVT);
 
       // Use a unique index in the instruction to prevent instruction folding.
       // Yes, this is a hack.
@@ -430,3 +414,9 @@ PTXTargetLowering::LowerCall(SDValue Chain, SDValue Callee,
 
   return Chain;
 }
+
+unsigned PTXTargetLowering::getNumRegisters(LLVMContext &Context, EVT VT) {
+  // All arguments consist of one "register," regardless of the type.
+  return 1;
+}
+