parse_cubin.cmake
上传用户:chinafayin
上传日期:2022-04-05
资源大小:153k
文件大小:4k
源码类别:

并行计算

开发平台:

Visual C++

  1. #  For more information, please see: http://software.sci.utah.edu
  2. #
  3. #  The MIT License
  4. #
  5. #  Copyright (c) 2007
  6. #  Scientific Computing and Imaging Institute, University of Utah
  7. #
  8. #  License for the specific language governing rights and limitations under
  9. #  Permission is hereby granted, free of charge, to any person obtaining a
  10. #  copy of this software and associated documentation files (the "Software"),
  11. #  to deal in the Software without restriction, including without limitation
  12. #  the rights to use, copy, modify, merge, publish, distribute, sublicense,
  13. #  and/or sell copies of the Software, and to permit persons to whom the
  14. #  Software is furnished to do so, subject to the following conditions:
  15. #
  16. #  The above copyright notice and this permission notice shall be included
  17. #  in all copies or substantial portions of the Software.
  18. #
  19. #  THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
  20. #  OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  21. #  FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
  22. #  THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  23. #  LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  24. #  FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
  25. #  DEALINGS IN THE SOFTWARE.
  26. # .cubin Parsing CMake Script
  27. # Abe Stephens
  28. # (c) 2007 Scientific Computing and Imaging Institute, University of Utah
  29. FILE(READ ${input_file} file_text)
  30. IF (${file_text} MATCHES ".+")
  31.   # Remember, four backslashes is escaped to one backslash in the string.
  32.   STRING(REGEX REPLACE ";" "\\;" file_text ${file_text})
  33.   STRING(REGEX REPLACE "ncode" ";code" file_text ${file_text})
  34.   
  35.   LIST(LENGTH file_text len)
  36.   FOREACH(line ${file_text})
  37.     # Only look at "code { }" blocks.
  38.     IF(line MATCHES "^code")
  39.       
  40.       # Break into individual lines.
  41.       STRING(REGEX REPLACE "n" ";" line ${line})
  42.       FOREACH(entry ${line})
  43.         # Extract kernel names.
  44.         IF (${entry} MATCHES "[^g]name = ([^ ]+)")
  45.           STRING(REGEX REPLACE ".* = ([^ ]+)" "\1" entry ${entry})
  46.           # Check to see if the kernel name starts with "_"
  47.           SET(skip FALSE)
  48.           # IF (${entry} MATCHES "^_")
  49.             # Skip the rest of this block.
  50.             # MESSAGE("Skipping ${entry}")
  51.             # SET(skip TRUE)
  52.           # ELSE (${entry} MATCHES "^_")
  53.             MESSAGE("Kernel:    ${entry}")  
  54.           # ENDIF (${entry} MATCHES "^_")
  55.         ENDIF(${entry} MATCHES "[^g]name = ([^ ]+)")
  56.         # Skip the rest of the block if necessary
  57.         IF(NOT skip)
  58.           # Registers
  59.           IF (${entry} MATCHES "reg = ([^ ]+)")
  60.             STRING(REGEX REPLACE ".* = ([^ ]+)" "\1" entry ${entry})
  61.             MESSAGE("Registers: ${entry}")
  62.           ENDIF(${entry} MATCHES "reg = ([^ ]+)")
  63.           
  64.           # Local memory
  65.           IF (${entry} MATCHES "lmem = ([^ ]+)")
  66.             STRING(REGEX REPLACE ".* = ([^ ]+)" "\1" entry ${entry})
  67.             MESSAGE("Local:     ${entry}")
  68.           ENDIF(${entry} MATCHES "lmem = ([^ ]+)")
  69.           
  70.           # Shared memory
  71.           IF (${entry} MATCHES "smem = ([^ ]+)")
  72.             STRING(REGEX REPLACE ".* = ([^ ]+)" "\1" entry ${entry})
  73.             MESSAGE("Shared:    ${entry}")
  74.           ENDIF(${entry} MATCHES "smem = ([^ ]+)")
  75.                   
  76.           IF (${entry} MATCHES "^}")
  77.             MESSAGE("")
  78.           ENDIF(${entry} MATCHES "^}")
  79.         ENDIF(NOT skip)
  80.       ENDFOREACH(entry)
  81.     ENDIF(line MATCHES "^code")
  82.   ENDFOREACH(line) 
  83. ELSE()
  84.   # MESSAGE("FOUND NO DEPENDS")
  85. ENDIF()