ext/torch/ext.cpp in torch-rb-0.8.0 vs ext/torch/ext.cpp in torch-rb-0.8.1

@@ -1,13 +1,17 @@
 #include <torch/torch.h>
 
 #include <rice/rice.hpp>
 
+void init_fft(Rice::Module& m);
+void init_linalg(Rice::Module& m);
 void init_nn(Rice::Module& m);
+void init_special(Rice::Module& m);
 void init_tensor(Rice::Module& m, Rice::Class& c, Rice::Class& rb_cTensorOptions);
 void init_torch(Rice::Module& m);
 
+void init_backends(Rice::Module& m);
 void init_cuda(Rice::Module& m);
 void init_device(Rice::Module& m);
 void init_ivalue(Rice::Module& m, Rice::Class& rb_cIValue);
 void init_random(Rice::Module& m);
 
@@ -25,10 +29,14 @@
 
   // keep this order
   init_torch(m);
   init_tensor(m, rb_cTensor, rb_cTensorOptions);
   init_nn(m);
+  init_fft(m);
+  init_linalg(m);
+  init_special(m);
 
+  init_backends(m);
   init_cuda(m);
   init_device(m);
   init_ivalue(m, rb_cIValue);
   init_random(m);
 }