adding setFlag and clearFlag for storages

zhouyonglong · Dec 19, 2015 · 6335106 · 6335106
1 parent fe768d7
commit 6335106
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/THCStorage.c b/THCStorage.c
@@ -119,6 +119,16 @@ THCudaStorage* THCudaStorage_newWithData(THCState *state, float *data, long size
   return storage;
 }
 
+void THCudaStorage_setFlag(THCState *state, THCudaStorage *storage, const char flag)
+{
+  storage->flag |= flag;
+}
+
+void THCudaStorage_clearFlag(THCState *state, THCudaStorage *storage, const char flag)
+{
+  storage->flag &= ~flag;
+}
+
 void THCudaStorage_retain(THCState *state, THCudaStorage *self)
 {
   if(self && (self->flag & TH_STORAGE_REFCOUNTED))