Initial commit

include/cpuCNN.cuh

@@ -0,0 +1,31 @@
+/* 
+ * File:   cpuFuncs.h
+ * Author: Alex Krizhevsky
+ *
+ * Created on September 10, 2012, 5:05 PM
+ */
+
+#ifndef CPUFUNCS_H
+#define	CPUFUNCS_H
+#include <helper_cuda.h>
+#include <softmaxtree.cuh>
+/*
+ * weights: (numNodes, numFeatures)
+ * nodes:   numNodesAtDepth-length array of ushort2 
+ *          where x coordinate gives node idx and y coordinate gives parent idx
+ * targets: (numNodes, numFeatures)
+ * 
+ */
+void cpuSoftmaxTreeFwd(float* weights, float* targets, const int numFeatures, SoftmaxTree& tree);
+
+/*
+ * grads:   (numNodes, numFeatures)
+ * 
+ */
+void cpuSoftmaxTreeBwd(float* grads, const int numFeatures, SoftmaxTree& tree);
+
+void cpuSoftmaxTreeUpdateWeights(float* weights, float* weightsInc, float* weightsGrad,
+                                 const int numFeatures, float eps, const float mom, float wc, SoftmaxTree& tree);
+
+#endif	/* CPUFUNCS_H */
+