#pragma once #include namespace torch::nativert { enum class OpKernelKind : uint8_t { kPrimKernel, kStaticDispatchKernel, kInterpreterFallbackKernel, // static dispatch kernels that don't re-use // out TensorImpl kNativeStaticDispatchKernel, }; } // namespace torch::nativert