mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
48a3349c29
This deletes most of the dead Tensor code paths, including the TensorMethods cwrap and generic/Tensor.cpp. This also moves the THNN.cwrap/.cpp generation to generate_code which can use ninja if installed.
29 lines
668 B
C++
29 lines
668 B
C++
#pragma once
|
|
|
|
#include <Python.h>
|
|
#include <memory>
|
|
#include <vector>
|
|
#include <ATen/ATen.h>
|
|
|
|
namespace torch {
|
|
|
|
struct TupleParser {
|
|
TupleParser(PyObject* args, int num_args=-1);
|
|
|
|
void parse(bool& x, const std::string& param_name);
|
|
void parse(int& x, const std::string& param_name);
|
|
void parse(double& x, const std::string& param_name);
|
|
void parse(std::vector<int>& x, const std::string& param_name);
|
|
void parse(std::string& x, const std::string& param_name);
|
|
|
|
protected:
|
|
std::runtime_error invalid_type(const std::string& expected, const std::string& param_name);
|
|
PyObject* next_arg();
|
|
|
|
private:
|
|
PyObject* args;
|
|
int idx;
|
|
};
|
|
|
|
} // namespace torch
|