"""Stable, byte-verified CUDA input paths across source and wheel installations.""" import hashlib,os,tempfile from pathlib import Path def stable_source(path,cache_root=None): path=Path(path) if path.suffix!='.cu':raise ValueError('CUDA source required') raw=path.read_bytes();digest=hashlib.sha256(raw).hexdigest() root=Path(cache_root) if cache_root is not None else Path(os.environ.get('TORCH_EXTENSIONS_DIR',str(Path.home()/'.cache/torch_extensions')))/'fidel_sources' folder=root/digest;folder.mkdir(parents=True,exist_ok=True);target=folder/path.name if not target.exists(): with tempfile.NamedTemporaryFile(dir=folder,delete=False) as stream: temporary=Path(stream.name);stream.write(raw);stream.flush();os.fsync(stream.fileno()) try: try:os.link(temporary,target) except FileExistsError:pass finally:temporary.unlink() if hashlib.sha256(target.read_bytes()).hexdigest()!=digest: raise RuntimeError('Cached CUDA source bytes differ; refusing compilation') os.utime(target,(946684800,946684800)) return target