Initial commit

include/multisoftmax.h

@@ -0,0 +1,38 @@
+/* 
+ * File:   multisoftmax.h
+ * Author: Alex Krizhevsky
+ *
+ * Created on May 9, 2012, 5:36 PM
+ */
+
+#ifndef MULTISOFTMAX_H
+#define	MULTISOFTMAX_H
+
+#include <algorithm>
+#include <thread.h>
+#include <matrix.h>
+#include <vector>
+
+#ifndef DIVUP
+#define DIVUP(x, y) (((x) + (y) - 1) / (y))
+#endif
+
+#define EXP exp
+#define LOG log
+#define INF 1e35f
+
+class MultiSoftmaxWorker : public Thread {
+protected:
+    Matrix* _elts, *_B, *_probs, *_fixed;
+    int _size;
+    bool _nofix;
+    void* run();
+public:
+    MultiSoftmaxWorker(Matrix* elts, Matrix* B, Matrix* probs, Matrix* _fixed, int size, bool nofix);
+    virtual ~MultiSoftmaxWorker();
+};
+
+void MultiSoftmaxCPU_T_parallel(Matrix& elts, std::vector<Matrix*>& B, Matrix& probs, Matrix& fixed, int size, bool nofix);
+
+#endif	/* MULTISOFTMAX_H */
+