daniel/h3-blackwell-runtime
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Fix pack profiler imports

Browse source
This commit is contained in:
Daniel Maddern 2026-08-15 02:57:21 +07:00
parent ddc0c5a2b7
commit 8d90cec133
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tools/profile_nvfp4_pack.py
View file

@ -5,10 +5,13 @@ from __future__ import annotations
import argparse
import json
import os
import sys
import time
import warnings
from pathlib import Path
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
warnings.filterwarnings("ignore", message="Found GPU0 NVIDIA GB10 which is of cuda capability 12.1.*", category=UserWarning)
import torch