TARGET_SPACE / tests /conftest.py
RDS777's picture
Upload 62 files
7dcbd04 verified
Raw History Blame Contribute Delete
348 Bytes
import os
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
try: # sans GPU, les noyaux Triton sont exécutés via l'interpréteur Triton
import torch
if not torch.cuda.is_available():
os.environ.setdefault("TRITON_INTERPRET", "1")
except ImportError: # pragma: no cover
pass