From 8d90cec1335e83d6c135d54a4267d1aaa16bb1c8 Mon Sep 17 00:00:00 2001 From: Daniel Maddern Date: Sat, 15 Aug 2026 02:57:21 +0700 Subject: [PATCH] Fix pack profiler imports --- tools/profile_nvfp4_pack.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tools/profile_nvfp4_pack.py b/tools/profile_nvfp4_pack.py index 02b2b25..4fc0636 100644 --- a/tools/profile_nvfp4_pack.py +++ b/tools/profile_nvfp4_pack.py @@ -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