Fix potential triton mission issue

This commit is contained in:
Mr. For Example
2025-06-05 13:47:53 +01:00
parent 30b7c8cdb2
commit 359e3b0e07
@@ -3,8 +3,6 @@ from abc import ABC, abstractmethod
import torch
import torch.nn.functional as F
import triton
import triton.language as tl
from torch.utils.cpp_extension import load_inline
from .utils import SINGLE_IMAGE_TYPE, image_to_tensor
@@ -112,6 +110,8 @@ class PBBackend(ABC):
try:
import triton
import triton.language as tl
@triton.jit
def pb_triton_step_kernel(