/
glebestraikh
/
adept
Обзор
Документация
Войти
/
glebestraikh
/
adept
Код
Запросы
0
Задачи
Вики
Пакеты
0
Релизы
0
Аналитика
Безопасность
master
python/tests/mptest.py
17 строк
484 B
kolkir
Add multiprocessing support for Tensor
13 авг 2025, 00:54
13 авг 2025, 00:54
92c362d
Код
Авторство
О чём код?
from adept import Tensor, TensorProperties, Shape, device_t, dtype_t from adept.multiprocessing import Pool import os def gen_tensor(): print(f"child pid: {os.getpid()}") props = TensorProperties(Shape([1, 1]), device_t.CPU, dtype_t.Int32, 0) return Tensor([os.getpid()], props) if __name__ == "__main__": print(f"main pid: {os.getpid()}") with Pool(processes=4) as pool: result = pool.apply_async(gen_tensor) t = result.get() print(t)