File size: 296 Bytes
33d23e1 | 1 2 3 4 5 6 7 8 9 10 11 12 13 | # SPDX-FileCopyrightText: © 2026 Tenstorrent USA, Inc.
# SPDX-License-Identifier: Apache-2.0
import os
import torch
from loguru import logger
num_torch_threads = max(1, os.cpu_count())
logger.info(f"Setting torch num_threads to {num_torch_threads}")
torch.set_num_threads(num_torch_threads)
|