Skip to content

Commit e8dc443

Browse files
committed
fix(compiler.py): delete nvprof_nvtx_pop()
1 parent e3e99ac commit e8dc443

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

python/paddle/incubate/fp8/deep_gemm/jit/compiler.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,6 @@
2323
import subprocess
2424
import uuid
2525

26-
import paddle
27-
2826
from ..utils import get_cuda_home
2927
from . import interleave_ffma
3028
from .runtime import Runtime, RuntimeCache
@@ -158,7 +156,6 @@ def build(name: str, arg_defs: tuple, code: str) -> Runtime:
158156
if os.getenv("DG_JIT_DEBUG", None):
159157
print(f"Using cached JIT runtime {name} during build")
160158
return runtime_cache[path]
161-
paddle.base.core.nvprof_nvtx_pop()
162159
# Write the code
163160
os.makedirs(path, exist_ok=True)
164161
args_path = f"{path}/kernel.args"

0 commit comments

Comments
 (0)