torch/csrc/cuda/AutoGPU.h

#ifndef THCP_AUTOGPU_INC
#define THCP_AUTOGPU_INC

#include <Python.h>
#include "torch/csrc/utils/auto_gpu.h"

class THCPAutoGPU : public AutoGPU {
public:
  explicit THCPAutoGPU(int device_id=-1);
  THCPAutoGPU(PyObject *args, PyObject *self=NULL);
  void setObjDevice(PyObject *obj);
};

#endif