(lengthx, lengthy, lengthz, exe, device)
| 126 | return max_mem_available_in_bytes(exe, device) / (numbers_in_one_datapoint * bytes_in_one_number) |
| 127 | |
| 128 | def maxBatchSize(lengthx, lengthy, lengthz, exe, device): |
| 129 | problemSize = int(lengthx) * int(lengthy) * int(lengthz) |
| 130 | maxBatchSize = max_problem_size(exe, device) / problemSize |
| 131 | if int(lengthx) == pow(2,16) or int(lengthx) == pow(2,17): |
| 132 | # special cases in the kernel. extra padding is added in, so we need to shrink the batch size to accommodate |
| 133 | return str(maxBatchSize/2) |
| 134 | else: |
| 135 | return str(maxBatchSize) |
| 136 | |
| 137 | def create_ini_file_if_requested(args): |
| 138 | if args.createIniFilename: |
nothing calls this directly
no test coverage detected