1#pragma once
2
3#include <torch/all.h>
4
5#ifdef TORCH_API_INCLUDE_EXTENSION_H
6#include <torch/extension.h>
7
8#endif // defined(TORCH_API_INCLUDE_EXTENSION_H)
9