Почему создание последовательности Соболя в SciPy зависает после завершения 25% последовательности? ⇐ Python

[Anonymous]

Я выполняю тяжелое моделирование методом квази-Монте-Карло с использованием последовательности Соболя, мне нужен генератор, который сможет создавать более 2 ** 30 выборок. SciPy говорит, что генератор ограничен значением 2 ** 30.
В нем также говорится, что, изменяя биты аргументов, можно поднять ограничение до максимума 2 ** 64 (для меня этого достаточно).

Код: Выделить всё

    sampler = qmc.Sobol(d=2, scramble=True, bits=64)

Однако я все еще сталкиваюсь с той же ошибкой: ровно на 25 процентах функции все останавливается без сообщения об ошибке. Я уверен, что проблема в генераторе, потому что я пытался переместить его внутри цикла for, и это сработало, однако повторное создание последовательности Соболя каждый раз, как я думаю, может вызвать математические проблемы со свойствами QMC.
Вот это код в минимальной воспроизводимой версии

Код: Выделить всё

from scipy.stats import qmc
from tqdm import tqdm

def run_simulation_batch_cupy_QMC(samples, a):
batch_size = 2 ** 27
n_batches = samples // batch_size
sampler = qmc.Sobol(d=2, scramble=True, bits=64)

for i in tqdm(range(n_batches)):

qmc_samples = sampler.random(n=batch_size)

#Here there is some computation I cut off

if __name__ == "__main__":

samples = 2 ** 34             #Samples must be, to get the bug, greater than 10 ** 10 and a power of 2 so 2**34
a_values = [1.1, 1.2, 1.3]      #Example

for a in tqdm(a_values):
run_simulation_batch_cupy_QMC(samples, a)

Я ожидал, что функция завершит генерацию. Мониторинг графического процессора и прочего могу сказать, что переполнения памяти не наблюдаю. Графический процессор просто прекращает вычисления ровно на 25 %.


Подробнее здесь: https://stackoverflow.com/questions/797 ... -25-of-the