Functions
	merge_files_into_workarounds (files)

	write_header (filename, workarounds)

	main (argv)

Variables
str	_LICENSE

tuple	_DO_NOT_EDIT_WARNING

Detailed Description

code generator for gpu workaround definitions

Function Documentation

◆ main()

tools.build_workaround_header.main ( argv )

Definition at line 54 of file build_workaround_header.py.

def main(argv):
  usage = "usage: %prog [options] file1 file2 file3 etc"
  parser = OptionParser(usage=usage)
  parser.add_option(
      "--output-file",
      dest="output_file",
      default="gpu_driver_bug_workaround_autogen.h",
      help="the name of the header file to write")
 
  (options, _) = parser.parse_args(args=argv)
 
  workarounds = merge_files_into_workarounds(parser.largs)
  write_header(options.output_file, workarounds)
 
 

◆ merge_files_into_workarounds()

tools.build_workaround_header.merge_files_into_workarounds ( files )

Definition at line 22 of file build_workaround_header.py.

def merge_files_into_workarounds(files):
  workarounds = set()
  for filename in files:
    with open(filename, 'r') as f:
      workarounds.update([workaround.strip() for workaround in f])
  return sorted(list(workarounds))
 
 

◆ write_header()

tools.build_workaround_header.write_header	(	filename,
		workarounds
	)

Definition at line 30 of file build_workaround_header.py.

def write_header(filename, workarounds):
  max_workaround_len = len(max(workarounds, key=len))
 
  with open(filename, 'w') as f:
    f.write(_LICENSE)
    f.write(_DO_NOT_EDIT_WARNING)
 
    indent = '  '
    macro = 'GPU_OP'
 
    # length of max string passed to write + 1
    max_len = len(indent) + len(macro) + 1 + max_workaround_len + 1 + 1
    write = lambda line: f.write(line + ' ' * (max_len - len(line)) + '\\\n')
    f.write('#ifndef GPU_DRIVER_BUG_WORKAROUNDS\n')
    write('#define GPU_DRIVER_BUG_WORKAROUNDS(GPU_OP)')
    for w in workarounds:
      write(indent + macro + '(' + w.upper() + ',')
      write(indent + ' ' * (len(macro) + 1) + w + ')')
 
    # one extra line for the last escaped newline to handle.
    f.write('\n')
    f.write('#endif  // GPU_DRIVER_BUG_WORKAROUNDS\n')
 
 

Variable Documentation

◆ _DO_NOT_EDIT_WARNING

tuple tools.build_workaround_header._DO_NOT_EDIT_WARNING

protected

Initial value:

=  ("// This file is auto-generated from " +
                        os.path.basename(__file__) + "\n" +
                        "// DO NOT EDIT!\n\n")

Definition at line 18 of file build_workaround_header.py.

◆ _LICENSE

str tools.build_workaround_header._LICENSE