mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
4f72d932fe
Summary: original pr - https://github.com/pytorch/pytorch/pull/161798 Test Plan: ci Rollback Plan: Differential Revision: D81724234 Pull Request resolved: https://github.com/pytorch/pytorch/pull/162217 Approved by: https://github.com/SherlockNoMad
18 lines
318 B
C++
18 lines
318 B
C++
#pragma once
|
|
|
|
#include <cstdint>
|
|
|
|
namespace torch::nativert {
|
|
|
|
enum class OpKernelKind : uint8_t {
|
|
kPrimKernel,
|
|
kStaticDispatchKernel,
|
|
kInterpreterFallbackKernel,
|
|
// static dispatch kernels that don't reuse
|
|
// out TensorImpl
|
|
kNativeStaticDispatchKernel,
|
|
kTritonKernel,
|
|
};
|
|
|
|
} // namespace torch::nativert
|