cmake/FindCUDA/parse_cubin.cmake

   1 #  James Bigler, NVIDIA Corp (nvidia.com - jbigler)
   2 #  Abe Stephens, SCI Institute -- http://www.sci.utah.edu/~abe/FindCuda.html
   3 #
   4 #  Copyright (c) 2008 - 2009 NVIDIA Corporation.  All rights reserved.
   5 #
   6 #  Copyright (c) 2007-2009
   7 #  Scientific Computing and Imaging Institute, University of Utah
   8 #
   9 #  This code is licensed under the MIT License.  See the FindCUDA.cmake script
  10 #  for the text of the license.
  11
  12 # The MIT License
  13 #
  14 # License for the specific language governing rights and limitations under
  15 # Permission is hereby granted, free of charge, to any person obtaining a
  16 # copy of this software and associated documentation files (the "Software"),
  17 # to deal in the Software without restriction, including without limitation
  18 # the rights to use, copy, modify, merge, publish, distribute, sublicense,
  19 # and/or sell copies of the Software, and to permit persons to whom the
  20 # Software is furnished to do so, subject to the following conditions:
  21 #
  22 # The above copyright notice and this permission notice shall be included
  23 # in all copies or substantial portions of the Software.
  24 #
  25 # THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  26 # OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  27 # FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  28 # THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  29 # LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  30 # FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  31 # DEALINGS IN THE SOFTWARE.
  32 #
  33
  34 #######################################################################
  35 # Parses a .cubin file produced by nvcc and reports statistics about the file.
  36
  37
  38 file(READ ${input_file} file_text)
  39
  40 if (${file_text} MATCHES ".+")
  41
  42   # Remember, four backslashes is escaped to one backslash in the string.
  43   string(REGEX REPLACE ";" "\\\\;" file_text ${file_text})
  44   string(REGEX REPLACE "\ncode" ";code" file_text ${file_text})
  45
  46   list(LENGTH file_text len)
  47
  48   foreach(line ${file_text})
  49
  50     # Only look at "code { }" blocks.
  51     if(line MATCHES "^code")
  52
  53       # Break into individual lines.
  54       string(REGEX REPLACE "\n" ";" line ${line})
  55
  56       foreach(entry ${line})
  57
  58         # Extract kernel names.
  59         if (${entry} MATCHES "[^g]name = ([^ ]+)")
  60           string(REGEX REPLACE ".* = ([^ ]+)" "\\1" entry ${entry})
  61
  62           # Check to see if the kernel name starts with "_"
  63           set(skip FALSE)
  64           # if (${entry} MATCHES "^_")
  65             # Skip the rest of this block.
  66             # message("Skipping ${entry}")
  67             # set(skip TRUE)
  68           # else (${entry} MATCHES "^_")
  69             message("Kernel:    ${entry}")
  70           # endif (${entry} MATCHES "^_")
  71
  72         endif(${entry} MATCHES "[^g]name = ([^ ]+)")
  73
  74         # Skip the rest of the block if necessary
  75         if(NOT skip)
  76
  77           # Registers
  78           if (${entry} MATCHES "reg([ ]+)=([ ]+)([^ ]+)")
  79             string(REGEX REPLACE ".*([ ]+)=([ ]+)([^ ]+)" "\\3" entry ${entry})
  80             message("Registers: ${entry}")
  81           endif()
  82
  83           # Local memory
  84           if (${entry} MATCHES "lmem([ ]+)=([ ]+)([^ ]+)")
  85             string(REGEX REPLACE ".*([ ]+)=([ ]+)([^ ]+)" "\\3" entry ${entry})
  86             message("Local:     ${entry}")
  87           endif()
  88
  89           # Shared memory
  90           if (${entry} MATCHES "smem([ ]+)=([ ]+)([^ ]+)")
  91             string(REGEX REPLACE ".*([ ]+)=([ ]+)([^ ]+)" "\\3" entry ${entry})
  92             message("Shared:    ${entry}")
  93           endif()
  94
  95           if (${entry} MATCHES "^}")
  96             message("")
  97           endif()
  98
  99         endif(NOT skip)
 100
 101
 102       endforeach(entry)
 103
 104     endif(line MATCHES "^code")
 105
 106   endforeach(line)
 107
 108 else()
 109   # message("FOUND NO DEPENDS")
 110 endif()
 111
 112