mark_contiguous operation

Author: Martin Raison

torch/csrc/generic/methods/SparseTensor.cwrap

@@ -94,6 +94,15 @@ PyObject * THSPTensor_(size)(PyObject *self, PyObject *args, PyObject *kwargs)
  - THSTensor* self
 ]]
 
+[[
+ name: markContiguous
+ python_name: mark_contiguous
+ sparse: yes
+ return: argument 0
+ arguments:
+ - THSTensor* self
+]]
+
 [[
  name: clone
  sparse: yes

torch/lib/THCS/generic/THCSTensor.cu

@@ -119,6 +119,10 @@ void THCSTensor_(contiguous)(THCState *state, THCSTensor *self) {
  self->contiguous = 1;
 }
 
+void THCSTensor_(markContiguous)(THCState *state, THCSTensor *self) {
+ self->contiguous = 1;
+}
+
 // In place transpose
 void THCSTensor_(transpose)(THCState *state, THCSTensor *self, int d1, int d2) {
  /* TODO

torch/lib/THCS/generic/THCSTensor.h

@@ -60,6 +60,7 @@ TH_API void THCSTensor_(copy)(THCState *state, THCSTensor *self, THCSTensor *src
 TH_API void THCSTensor_(transpose)(THCState *state, THCSTensor *self, int dimension1_, int dimension2_);
 TH_API int THCSTensor_(isContiguous)(THCState *state, const THCSTensor *self);
 TH_API void THCSTensor_(contiguous)(THCState *state, THCSTensor *self);
+TH_API void THCSTensor_(markContiguous)(THCState *state, THCSTensor *self);
 
 TH_API void THCTensor_(sparseMask)(THCState *state, THCSTensor *r_, THCTensor *t, THCSTensor *mask);
 

torch/lib/THS/generic/THSTensor.c

@@ -508,6 +508,10 @@ void THSTensor_(contiguous)(THSTensor *self) {
  self->contiguous = 1;
 }
 
+void THSTensor_(markContiguous)(THSTensor *self) {
+ self->contiguous = 1;
+}
+
 void THTensor_(sparseMask)(THSTensor *r_, THTensor *t, THSTensor *mask) {
  THSTensor_(resizeAs)(r_, mask);
  if (mask->nnz == 0) {

torch/lib/THS/generic/THSTensor.h

@@ -59,6 +59,7 @@ TH_API void THSTensor_(transpose)(THSTensor *self, int dimension1_, int dimensio
 TH_API int THSTensor_(isContiguous)(const THSTensor *self);
 TH_API int THSTensor_(isSameSizeAs)(const THSTensor *self, const THSTensor *src);
 TH_API void THSTensor_(contiguous)(THSTensor *self);
+TH_API void THSTensor_(markContiguous)(THSTensor *self);
 
 TH_API void THTensor_(sparseMask)(THSTensor *r_, THTensor *t, THSTensor *mask);