(exe, device)
| 121 | return int(maxMemoryAvailable.group(0)) |
| 122 | |
| 123 | def max_problem_size(exe, device): |
| 124 | numbers_in_one_datapoint = 2 # (i.e.: real or complex?) |
| 125 | bytes_in_one_number = 4 # (i.e.: single or double precision?) |
| 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) |
no test coverage detected