Source code for stgraph.compiler.backend.pytorch.torch_kernel_wrapper

import torch
from stgraph.compiler.backend.kernel_wrapper import KernelWrapper

[docs]class KernelWrapperTorch(KernelWrapper, torch.autograd.Function):
    def __init__(self):
        super().__init__()