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 argparse
import json import json
import os import os
import sys
import time import time
import warnings import warnings
from pathlib import Path 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) warnings.filterwarnings("ignore", message="Found GPU0 NVIDIA GB10 which is of cuda capability 12.1.*", category=UserWarning)
import torch import torch