http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.10/src/test/scala/org/apache/mahout/viennacl/opencl/ViennaCLSuiteVCL.scala ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.10/src/test/scala/org/apache/mahout/viennacl/opencl/ViennaCLSuiteVCL.scala b/viennacl/scala-2.10/src/test/scala/org/apache/mahout/viennacl/opencl/ViennaCLSuiteVCL.scala new file mode 100644 index 0000000..73787e3 --- /dev/null +++ b/viennacl/scala-2.10/src/test/scala/org/apache/mahout/viennacl/opencl/ViennaCLSuiteVCL.scala @@ -0,0 +1,441 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.mahout.opencl.viennacl + +import org.apache.mahout.math._ +import org.apache.mahout.math.scalabindings.RLikeOps._ +import org.apache.mahout.viennacl.opencl.javacpp.CompressedMatrix +import org.apache.mahout.viennacl.opencl._ +import org.apache.mahout.viennacl.opencl.javacpp.Functions._ +import org.apache.mahout.viennacl.opencl.javacpp.LinalgFunctions._ +import org.apache.mahout.viennacl.opencl.javacpp.{Context, LinalgFunctions, VCLVector, _} +import org.bytedeco.javacpp.DoublePointer +import org.scalatest.{FunSuite, Matchers} + +import scala.util.Random + +class ViennaCLSuiteVCL extends FunSuite with Matchers { + + test("row-major viennacl::matrix") { + + // Just to make sure the javacpp library is loaded: + Context.loadLib() + + val m = 20 + val n = 30 + val data = new DoublePointer(m * n) + val buff = data.asBuffer() + // Fill with some noise + while (buff.remaining() > 0) buff.put(Random.nextDouble()) + + // Create row-major matrix with OpenCL + val openClCtx = new Context(Context.OPENCL_MEMORY) + val hostClCtx = new Context(Context.MAIN_MEMORY) + val oclMx = new DenseRowMatrix(m, n, openClCtx) + val cpuMx = new DenseRowMatrix(data = data, nrow = m, ncol = n, hostClCtx) + + oclMx.memoryDomain shouldBe Context.OPENCL_MEMORY + + // Apparently, this doesn't really switch any contexts? any how, uncommenting this causes + // subsequent out-of-resources OCL error for me in other tests. Perhaps we shouldnt' really + // do cross-memory-domain assigns? + + // oclMx := cpuMx + + // Did it change memory domain? that may explain the OCL resource leak. + info(s"OCL matrix memory domain after assgn=${oclMx.memoryDomain}") + oclMx.memoryDomain shouldBe Context.OPENCL_MEMORY + + + // And free. + cpuMx.close() + oclMx.close() + + } + + test("dense vcl mmul with fast_copy") { + + import LinalgFunctions._ + + val vclCtx = new Context(Context.OPENCL_MEMORY) + + val m = 20 + val n = 30 + val s = 40 + + val r = new Random(1234) + + // Dense row-wise + val mxA = new DenseMatrix(m, s) + val mxB = new DenseMatrix(s, n) + + // add some data + mxA := { (_, _, _) => r.nextDouble() } + mxB := { (_, _, _) => r.nextDouble() } + + // time Mahout MMul + // mxC = mxA %*% mxB via Mahout MMul + val mxCControl = mxA %*% mxB + + val vclA = toVclDenseRM(mxA, vclCtx) + val vclB = toVclDenseRM(mxB, vclCtx) + + val vclC = new DenseRowMatrix(prod(vclA, vclB)) + + val mxC = fromVclDenseRM(vclC) + + vclA.close() + vclB.close() + vclC.close() + + // So did we compute it correctly? + (mxC - mxA %*% mxB).norm / m / n should be < 1e-16 + + vclCtx.deallocate() + vclCtx.close() + + } + + test("mmul microbenchmark") { + val oclCtx = new Context(Context.OPENCL_MEMORY) + val memCtx = new Context(Context.MAIN_MEMORY) + + val m = 3000 + val n = 3000 + val s = 1000 + + val r = new Random(1234) + + // Dense row-wise + val mxA = new DenseMatrix(m, s) + val mxB = new DenseMatrix(s, n) + + // add some data + mxA := { (_, _, _) => r.nextDouble() } + mxB := { (_, _, _) => r.nextDouble() } + + var ms = System.currentTimeMillis() + mxA %*% mxB + ms = System.currentTimeMillis() - ms + info(s"Mahout multiplication time: $ms ms.") + + import LinalgFunctions._ + + // openCL time, including copying: + { + ms = System.currentTimeMillis() + val oclA = toVclDenseRM(mxA, oclCtx) + val oclB = toVclDenseRM(mxB, oclCtx) + val oclC = new DenseRowMatrix(prod(oclA, oclB)) + val mxC = fromVclDenseRM(oclC) + ms = System.currentTimeMillis() - ms + info(s"ViennaCL/OpenCL multiplication time: $ms ms.") + + oclA.close() + oclB.close() + oclC.close() + } + + // openMP/cpu time, including copying: + { + ms = System.currentTimeMillis() + val ompA = toVclDenseRM(mxA, memCtx) + val ompB = toVclDenseRM(mxB, memCtx) + val ompC = new DenseRowMatrix(prod(ompA, ompB)) + val mxC = fromVclDenseRM(ompC) + ms = System.currentTimeMillis() - ms + info(s"ViennaCL/cpu/OpenMP multiplication time: $ms ms.") + + ompA.close() + ompB.close() + ompC.close() + } + oclCtx.deallocate() + oclCtx.close() + + + } + + test("trans") { + + val oclCtx = new Context(Context.OPENCL_MEMORY) + val ompCtx = new Context(Context.MAIN_MEMORY) + + + val m = 20 + val n = 30 + + val r = new Random(1234) + + // Dense row-wise + val mxA = new DenseMatrix(m, n) + + // add some data + mxA := { (_, _, _) => r.nextDouble() } + + // Test transposition in OpenCL + { + val oclA = toVclDenseRM(src = mxA, oclCtx) + val oclAt = new DenseRowMatrix(trans(oclA)) + + val mxAt = fromVclDenseRM(oclAt) + oclA.close() + oclAt.close() + + (mxAt - mxA.t).norm / m / n should be < 1e-16 + } + + // Test transposition in OpenMP + { + val ompA = toVclDenseRM(src = mxA, ompCtx) + val ompAt = new DenseRowMatrix(trans(ompA)) + + val mxAt = fromVclDenseRM(ompAt) + ompA.close() + ompAt.close() + + (mxAt - mxA.t).norm / m / n should be < 1e-16 + } + oclCtx.deallocate() + oclCtx.close() + + + } + + test("sparse mmul microbenchmark") { + + val oclCtx = new Context(Context.OPENCL_MEMORY) + val ompCtx = new Context(Context.MAIN_MEMORY) + + + val m = 3000 + val n = 3000 + val s = 1000 + + val r = new Random(1234) + + // sparse row-wise + val mxA = new SparseRowMatrix(m, s, false) + val mxB = new SparseRowMatrix(s, n, true) + + // add some sparse data with a 20% threshold + mxA := { (_, _, v) => if (r.nextDouble() < .20) r.nextDouble() else v } + mxB := { (_, _, v) => if (r.nextDouble() < .20) r.nextDouble() else v } + + var ms = System.currentTimeMillis() + val mxC = mxA %*% mxB + ms = System.currentTimeMillis() - ms + info(s"Mahout Sparse multiplication time: $ms ms.") + +// Test multiplication in OpenCL + { + + ms = System.currentTimeMillis() + val oclA = toVclCmpMatrixAlt(mxA, oclCtx) + val oclB = toVclCmpMatrixAlt(mxB, oclCtx) + + val oclC = new CompressedMatrix(prod(oclA, oclB)) + ms = System.currentTimeMillis() - ms + info(s"ViennaCL/OpenCL Sparse multiplication time: $ms ms.") + + val oclMxC = fromVclCompressedMatrix(oclC) + val ompMxC = fromVclCompressedMatrix(oclC) + (mxC - oclMxC).norm / mxC.nrow / mxC.ncol should be < 1e-16 + + oclA.close() + oclB.close() + oclC.close() + } + + // Test multiplication in OpenMP + { + ms = System.currentTimeMillis() + // val ompA = toVclCompressedMatrix(src = mxA, ompCtx) + // val ompB = toVclCompressedMatrix(src = mxB, ompCtx) + + val ompA = toVclCmpMatrixAlt(mxA, ompCtx) + val ompB = toVclCmpMatrixAlt(mxB, ompCtx) + + val ompC = new CompressedMatrix(prod(ompA, ompB)) + + ms = System.currentTimeMillis() - ms + info(s"ViennaCL/cpu/OpenMP Sparse multiplication time: $ms ms.") + + val ompMxC = fromVclCompressedMatrix(ompC) + (mxC - ompMxC).norm / mxC.nrow / mxC.ncol should be < 1e-16 + + ompA.close() + ompB.close() + ompC.close() + + } + oclCtx.deallocate() + oclCtx.close() + + } + + test("VCL Dense Matrix %*% Dense vector") { + + val oclCtx = new Context(Context.OPENCL_MEMORY) + val mainCtx = new Context(Context.MAIN_MEMORY) + + + val m = 3000 + val s = 1000 + + val r = new Random(1234) + + // Dense row-wise + val mxA = new DenseMatrix(m, s) + val dvecB = new DenseVector(s) + + // add some random data + mxA := { (_,_,_) => r.nextDouble() } + dvecB := { (_,_) => r.nextDouble() } + + //test in matrix %*% vec + var ms = System.currentTimeMillis() + val mDvecC = mxA %*% dvecB + ms = System.currentTimeMillis() - ms + info(s"Mahout dense matrix %*% dense vector multiplication time: $ms ms.") + + + // Test mx %*% vec multiplication in OpenCL + { + + ms = System.currentTimeMillis() + + // we must first create vectors in main memory + // when working with vectors at least in ViennaCl + // this is the preferred method + val oclMxA = toVclDenseRM(mxA, mainCtx) + val oclVecB = toVclVec(dvecB, mainCtx) + + // now copy to the OpenCL device + oclMxA.switchMemoryContext(oclCtx) + oclVecB.switchMemoryContext(oclCtx) + + // perform multiplication + val oclVecC = new VCLVector(prod(oclMxA, oclVecB)) + + // copy back to main memory so that we may + // read values out of the result. This must be + // copied back to main memory VCL can not read + // directly from an OpenCL device + oclVecC.switchMemoryContext(mainCtx) + + val oclDvecC = fromVClVec(oclVecC) + + ms = System.currentTimeMillis() - ms + info(s"ViennaCL/OpenCL dense matrix %*% dense vector multiplication time: $ms ms.") + (oclDvecC.toColMatrix - mDvecC.toColMatrix).norm / s should be < 1e-10 + + oclMxA.close() + oclVecB.close() + oclVecC.close() + } + + //Test multiplication in OpenMP + { + + ms = System.currentTimeMillis() + val ompMxA = toVclDenseRM(mxA, mainCtx) + val ompVecB = toVclVec(dvecB, mainCtx) + + val ompVecC = new VCLVector(prod(ompMxA, ompVecB)) + val ompDvecC = fromVClVec(ompVecC) + + ms = System.currentTimeMillis() - ms + info(s"ViennaCL/cpu/OpenMP dense matrix %*% dense vector multiplication time: $ms ms.") + (ompDvecC.toColMatrix - mDvecC.toColMatrix).norm / s should be < 1e-10 + + ompMxA.close() + ompVecB.close() + ompVecC.close() + } + + oclCtx.deallocate() + oclCtx.close() + + } + + + test("Sparse %*% Dense mmul microbenchmark") { + val oclCtx = new Context(Context.OPENCL_MEMORY) + val memCtx = new Context(Context.MAIN_MEMORY) + + val m = 3000 + val n = 3000 + val s = 1000 + + val r = new Random(1234) + + // Dense row-wise + val mxSr = new SparseMatrix(m, s) + val mxDn = new DenseMatrix(s, n) + + // add some data + mxSr := { (_, _, v) => if (r.nextDouble() < .20) r.nextDouble() else v } + mxDn := { (_, _, _) => r.nextDouble() } + + var ms = System.currentTimeMillis() + mxSr %*% mxDn + ms = System.currentTimeMillis() - ms + info(s"Mahout multiplication time: $ms ms.") + + import LinalgFunctions._ + + // For now, since our dense matrix is fully dense lets just assume that our result is dense. + // openCL time, including copying: + { + ms = System.currentTimeMillis() + val oclA = toVclCmpMatrixAlt(mxSr, oclCtx) + val oclB = toVclDenseRM(mxDn, oclCtx) + val oclC = new DenseRowMatrix(prod(oclA, oclB)) + val mxC = fromVclDenseRM(oclC) + ms = System.currentTimeMillis() - ms + info(s"ViennaCL/OpenCL multiplication time: $ms ms.") + + oclA.close() + oclB.close() + oclC.close() + } + + // openMP/cpu time, including copying: + { + ms = System.currentTimeMillis() + val ompA = toVclCmpMatrixAlt(mxSr, memCtx) + val ompB = toVclDenseRM(mxDn, memCtx) + val ompC = new DenseRowMatrix(prod(ompA, ompB)) + val mxC = fromVclDenseRM(ompC) + ms = System.currentTimeMillis() - ms + info(s"ViennaCL/cpu/OpenMP multiplication time: $ms ms.") + + ompA.close() + ompB.close() + ompC.close() + } + + oclCtx.deallocate() + oclCtx.close() + + + } + + + +}
http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/linux-haswell.properties ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/linux-haswell.properties b/viennacl/scala-2.11/linux-haswell.properties new file mode 100644 index 0000000..3ee4494 --- /dev/null +++ b/viennacl/scala-2.11/linux-haswell.properties @@ -0,0 +1,28 @@ +platform=linux-x86_64 +platform.path.separator=: +platform.source.suffix=.cpp +platform.includepath.prefix=-I +platform.includepath= +platform.compiler=g++ +platform.compiler.cpp11=-std=c++11 +platform.compiler.default= +platform.compiler.fastfpu=-msse3 -ffast-math +platform.compiler.viennacl=-fopenmp -fpermissive +platform.compiler.nodeprecated=-Wno-deprecated-declarations +#build for haswell arch with for GCC >= 4.9.0 +platform.compiler.output=-Wl,-rpath,$ORIGIN/ -Wl,-z,noexecstack -Wl,-Bsymbolic -march=haswell -m64 -Wall -O3 -fPIC -shared -s -o\u0020 +#for GCC < 4.9.0 use -march=core-avx2 for haswell arch +#platform.compiler.output=-Wl,-rpath,$ORIGIN/ -Wl,-z,noexecstack -Wl,-Bsymbolic -march=core-avx2 -m64 -Wall -Ofast -fPIC -shared -s -o\u0020 +#build for native: +#platform.compiler.output=-Wl,-rpath,$ORIGIN/ -Wl,-z,noexecstack -Wl,-Bsymbolic -march=native -m64 -Wall -Ofast -fPIC -shared -s -o\u0020 +platform.linkpath.prefix=-L +platform.linkpath.prefix2=-Wl,-rpath, +platform.linkpath= +platform.link.prefix=-l +platform.link.suffix= +platform.link= +platform.framework.prefix=-F +platform.framework.suffix= +platform.framework= +platform.library.prefix=lib +platform.library.suffix=.so http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/linux-x86_64-viennacl.properties ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/linux-x86_64-viennacl.properties b/viennacl/scala-2.11/linux-x86_64-viennacl.properties new file mode 100644 index 0000000..e5de1fa --- /dev/null +++ b/viennacl/scala-2.11/linux-x86_64-viennacl.properties @@ -0,0 +1,24 @@ +platform=linux-x86_64 +platform.path.separator=: +platform.source.suffix=.cpp +platform.includepath.prefix=-I +platform.includepath= +platform.compiler=g++ +platform.compiler.cpp11=-std=c++11 +platform.compiler.default= +platform.compiler.fastfpu=-msse3 -ffast-math +platform.compiler.viennacl=-fopenmp -fpermissive +platform.compiler.nodeprecated=-Wno-deprecated-declarations +# platform.compiler.output=-Wl,-rpath,$ORIGIN/ -Wl,-z,noexecstack -Wl,-Bsymbolic -march=x86-64 -m64 -Wall -O3 -fPIC -shared -s -o\u0020 +platform.compiler.output=-Wl,-rpath,$ORIGIN/ -Wl,-z,noexecstack -Wl,-Bsymbolic -march=native -m64 -Wall -Ofast -fPIC -shared -s -o\u0020 +platform.linkpath.prefix=-L +platform.linkpath.prefix2=-Wl,-rpath, +platform.linkpath= +platform.link.prefix=-l +platform.link.suffix= +platform.link= +platform.framework.prefix=-F +platform.framework.suffix= +platform.framework= +platform.library.prefix=lib +platform.library.suffix=.so http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/pom.xml ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/pom.xml b/viennacl/scala-2.11/pom.xml index b874e22..d48f035 100644 --- a/viennacl/scala-2.11/pom.xml +++ b/viennacl/scala-2.11/pom.xml @@ -18,16 +18,20 @@ --> <project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance" xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/maven-v4_0_0.xsd"> - <modelVersion>4.0.0</modelVersion> - <parent> - <groupId>org.apache.mahout</groupId> - <artifactId>mahout-native-viennacl</artifactId> - <version>0.13.2-SNAPSHOT</version> - <relativePath>../pom.xml</relativePath> - </parent> + <modelVersion>4.0.0</modelVersion> - <artifactId>mahout-native-viennacl_2.11</artifactId> - <name>Mahout Native VienniaCL OpenCL Bindings for Scala 2.11 bindings</name> + <parent> + <groupId>org.apache.mahout</groupId> + <artifactId>mahout</artifactId> + <version>0.13.2-SNAPSHOT</version> + <relativePath>../pom.xml</relativePath> + </parent> + + <artifactId>mahout-native-viennacl_2.11</artifactId> + + <name>Mahout Native VienniaCL OpenCL Scala 2.11 Bindings</name> + <description>Native Structures and interfaces to be used from Mahout math-scala. + </description> <properties> @@ -35,33 +39,273 @@ <scala.version>2.11.8</scala.version> </properties> - <packaging>jar</packaging> + <packaging>jar</packaging> - <build> - <sourceDirectory>../src/main</sourceDirectory> - <plugins> + <build> + <plugins> + <!-- create test jar so other modules can reuse the native test utility classes. --> + <plugin> + <groupId>org.apache.maven.plugins</groupId> + <artifactId>maven-jar-plugin</artifactId> + <executions> + <execution> + <goals> + <goal>test-jar</goal> + </goals> + <phase>package</phase> + </execution> + </executions> + </plugin> - <plugin> - <artifactId>maven-antrun-plugin</artifactId> - <executions> - <execution> - <id>copy</id> - <phase>package</phase> - </execution> - </executions> - </plugin> - <!-- disable javacpp recompile, its wasteful and causes issues --> - <plugin> - <groupId>org.codehaus.mojo</groupId> - <artifactId>exec-maven-plugin</artifactId> - <executions> - <execution> - <id>javacpp</id> - <phase/> - </execution> - </executions> - </plugin> - </plugins> - </build> + <plugin> + <artifactId>maven-javadoc-plugin</artifactId> + </plugin> + + <plugin> + <artifactId>maven-source-plugin</artifactId> + </plugin> + + <plugin> + <groupId>net.alchim31.maven</groupId> + <artifactId>scala-maven-plugin</artifactId> + <executions> + <execution> + <id>add-scala-sources</id> + <phase>initialize</phase> + <goals> + <goal>add-source</goal> + </goals> + </execution> + <execution> + <id>scala-compile</id> + <phase>process-resources</phase> + <goals> + <goal>compile</goal> + </goals> + </execution> + <execution> + <id>scala-test-compile</id> + <phase>process-test-resources</phase> + <goals> + <goal>testCompile</goal> + </goals> + </execution> + </executions> + </plugin> + + <!--this is what scalatest recommends to do to enable scala tests --> + + <!-- disable surefire --> + <plugin> + <groupId>org.apache.maven.plugins</groupId> + <artifactId>maven-surefire-plugin</artifactId> + <configuration> + <skipTests>true</skipTests> + </configuration> + </plugin> + <!-- enable scalatest --> + <plugin> + <groupId>org.scalatest</groupId> + <artifactId>scalatest-maven-plugin</artifactId> + <executions> + <execution> + <id>test</id> + <goals> + <goal>test</goal> + </goals> + </execution> + </executions> + <configuration> + <argLine>-Xmx4g</argLine> + </configuration> + </plugin> + + + <!--JavaCPP native build plugin--> + <!-- old-style way to get it to compile. --> + <!--based on https://github.com/bytedeco/javacpp/wiki/Maven--> + <plugin> + <groupId>org.codehaus.mojo</groupId> + <artifactId>exec-maven-plugin</artifactId> + <version>1.2.1</version> + <executions> + <execution> + <id>javacpp</id> + <phase>process-classes</phase> + <goals> + <goal>exec</goal> + </goals> + <configuration> + <environmentVariables> + <LD_LIBRARY_PATH>{project.basedir}/target/classes/org/apache/mahout/javacpp/linalg/linux-x86_64/ + </LD_LIBRARY_PATH> + </environmentVariables> + <executable>java</executable> + <arguments> + <argument>-jar</argument> + <argument>${org.bytedeco:javacpp:jar}</argument> + <argument>-propertyfile</argument> + <argument>linux-x86_64-viennacl.properties</argument> + <argument>-classpath</argument> + <argument>${project.build.outputDirectory}:${org.scala-lang:scala-library:jar}</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.CompressedMatrix</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.Context</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.MatrixBase</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.DenseRowMatrix</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.DenseColumnMatrix</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.MatMatProdExpression</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.ProdExpression</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.SrMatDnMatProdExpression</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.MatrixTransExpression</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.LinalgFunctions</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.Functions</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.VectorBase</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.VCLVector</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.VecMultExpression</argument> + <argument>org.apache.mahout.viennacl.opencl.javacpp.MemHandle</argument> + <argument>org.apache.mahout.viennacl.opencl.GPUMMul</argument> + <argument>org.apache.mahout.viennacl.opencl.GPUMMul$</argument> + </arguments> + </configuration> + </execution> + </executions> + </plugin> + + <plugin> + <groupId>org.apache.maven.plugins</groupId> + <artifactId>maven-dependency-plugin</artifactId> + <version>2.3</version> + <executions> + <execution> + <goals> + <goal>properties</goal> + </goals> + </execution> + </executions> + </plugin> + <plugin> + <groupId>org.codehaus.mojo</groupId> + <artifactId>exec-maven-plugin</artifactId> + <version>1.2.1</version> + </plugin> + + <!-- copy jars to top directory, which is MAHOUT_HOME --> + <plugin> + <artifactId>maven-antrun-plugin</artifactId> + <version>1.4</version> + <executions> + <execution> + <id>copy</id> + <phase>package</phase> + <configuration> + <tasks> + <copy file="target/mahout-native-viennacl_2.11-${version}.jar" tofile="../../mahout-native-viennacl_2.11-${version}.jar" /> + </tasks> + </configuration> + <goals> + <goal>run</goal> + </goals> + </execution> + </executions> + </plugin> + <!-- delete jars on claen in top directory, which is MAHOUT_HOME --> + <plugin> + <artifactId>maven-clean-plugin</artifactId> + <version>3.0.0</version> + <configuration> + <filesets> + <fileset> + <directory>../</directory> + <includes> + <include>mahout-native-viennacl_2.11*.jar</include> + </includes> + <followSymlinks>false</followSymlinks> + </fileset> + </filesets> + </configuration> + </plugin> + </plugins> -</project> \ No newline at end of file + </build> + + <dependencies> + + <dependency> + <groupId>${project.groupId}</groupId> + <artifactId>mahout-math-scala_${scala.compat.version}</artifactId> + </dependency> + + <!-- 3rd-party --> + <dependency> + <groupId>log4j</groupId> + <artifactId>log4j</artifactId> + </dependency> + + <!-- scala stuff --> + <dependency> + <groupId>org.scalatest</groupId> + <artifactId>scalatest_${scala.compat.version}</artifactId> + </dependency> + + <dependency> + <groupId>org.bytedeco</groupId> + <artifactId>javacpp</artifactId> + <version>1.2.4</version> + </dependency> + + </dependencies> + + + <profiles> + <profile> + <id>mahout-release</id> + <build> + <plugins> + <plugin> + <groupId>net.alchim31.maven</groupId> + <artifactId>scala-maven-plugin</artifactId> + <executions> + <execution> + <id>generate-scaladoc</id> + <goals> + <goal>doc</goal> + </goals> + </execution> + <execution> + <id>attach-scaladoc-jar</id> + <goals> + <goal>doc-jar</goal> + </goals> + </execution> + </executions> + </plugin> + </plugins> + </build> + </profile> + <profile> + <id>travis</id> + <build> + <plugins> + <plugin> + <groupId>org.apache.maven.plugins</groupId> + <artifactId>maven-surefire-plugin</artifactId> + <configuration> + <!-- Limit memory for unit tests in Travis --> + <argLine>-Xmx4g</argLine> + <!--<argLine>-Djava.library.path=${project.build.directory}/libs/natives/linux-x86_64:${project.build.directory}/libs/natives/linux:${project.build.directory}/libs/natives/maxosx</argLine>--> + </configuration> + </plugin> + <plugin> + <groupId>org.apache.maven.plugins</groupId> + <artifactId>maven-failsafe-plugin</artifactId> + <configuration> + <!-- Limit memory for integration tests in Travis --> + <argLine>-Xmx4g</argLine> + <!--<argLine>-Djava.library.path=${project.build.directory}/libs/natives/linux-x86_64:${project.build.directory}/libs/natives/linux:${project.build.directory}/libs/natives/maxosx</argLine>--> + </configuration> + </plugin> + </plugins> + </build> + </profile> + </profiles> +</project> http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/java/org/apache/mahout/viennacl/opencl/javacpp/Functions.java ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/java/org/apache/mahout/viennacl/opencl/javacpp/Functions.java b/viennacl/scala-2.11/src/main/java/org/apache/mahout/viennacl/opencl/javacpp/Functions.java new file mode 100644 index 0000000..1c14f97 --- /dev/null +++ b/viennacl/scala-2.11/src/main/java/org/apache/mahout/viennacl/opencl/javacpp/Functions.java @@ -0,0 +1,104 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.mahout.viennacl.opencl.javacpp; + +import org.bytedeco.javacpp.BytePointer; +import org.bytedeco.javacpp.DoublePointer; +import org.bytedeco.javacpp.IntPointer; +import org.bytedeco.javacpp.annotation.*; + +import java.nio.DoubleBuffer; +import java.nio.IntBuffer; + + +@Properties(inherit = Context.class, + value = @Platform( + library = "jniViennaCL" + ) +) +@Namespace("viennacl") +public final class Functions { + + private Functions() { + } + + // This is (imo) an inconsistency in Vienna cl: almost all operations require MatrixBase, and + // fast_copy require type `matrix`, i.e., one of DenseRowMatrix or DenseColumnMatrix. + @Name("fast_copy") + public static native void fastCopy(DoublePointer srcBegin, DoublePointer srcEnd, @ByRef DenseRowMatrix dst); + + @Name("fast_copy") + public static native void fastCopy(DoublePointer srcBegin, DoublePointer srcEnd, @ByRef DenseColumnMatrix dst); + + @Name("fast_copy") + public static native void fastCopy(@ByRef DenseRowMatrix src, DoublePointer dst); + + @Name("fast_copy") + public static native void fastCopy(@ByRef DenseColumnMatrix src, DoublePointer dst); + + @Name("fast_copy") + public static native void fastCopy(@Const @ByRef VectorBase dst, @Const @ByRef VCLVector src); + + @Name("fast_copy") + public static native void fastCopy(@Const @ByRef VCLVector src, @Const @ByRef VectorBase dst); + + + @ByVal + public static native MatrixTransExpression trans(@ByRef MatrixBase src); + + @Name("backend::memory_read") + public static native void memoryReadInt(@Const @ByRef MemHandle src_buffer, + int bytes_to_read, + int offset, + IntPointer ptr, + boolean async); + + @Name("backend::memory_read") + public static native void memoryReadDouble(@Const @ByRef MemHandle src_buffer, + int bytes_to_read, + int offset, + DoublePointer ptr, + boolean async); + + @Name("backend::memory_read") + public static native void memoryReadInt(@Const @ByRef MemHandle src_buffer, + int bytes_to_read, + int offset, + IntBuffer ptr, + boolean async); + + @Name("backend::memory_read") + public static native void memoryReadDouble(@Const @ByRef MemHandle src_buffer, + int bytes_to_read, + int offset, + DoubleBuffer ptr, + boolean async); + + @Name("backend::memory_read") + public static native void memoryReadBytes(@Const @ByRef MemHandle src_buffer, + int bytes_to_read, + int offset, + BytePointer ptr, + boolean async); + + + static { + Context.loadLib(); + } + +} http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/java/org/apache/mahout/viennacl/opencl/javacpp/LinalgFunctions.java ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/java/org/apache/mahout/viennacl/opencl/javacpp/LinalgFunctions.java b/viennacl/scala-2.11/src/main/java/org/apache/mahout/viennacl/opencl/javacpp/LinalgFunctions.java new file mode 100644 index 0000000..9540691 --- /dev/null +++ b/viennacl/scala-2.11/src/main/java/org/apache/mahout/viennacl/opencl/javacpp/LinalgFunctions.java @@ -0,0 +1,86 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.mahout.viennacl.opencl.javacpp; + +import org.bytedeco.javacpp.annotation.*; + + +@Properties(inherit = Context.class, + value = @Platform( + library = "jniViennaCL" + ) +) +@Namespace("viennacl::linalg") +public final class LinalgFunctions { + + private LinalgFunctions() { + } + + static { + Context.loadLib(); + } + + + @ByVal + public static native MatMatProdExpression prod(@Const @ByRef MatrixBase a, + @Const @ByRef MatrixBase b); + + @ByVal + public static native ProdExpression prod(@Const @ByRef CompressedMatrix a, + @Const @ByRef CompressedMatrix b); + + @ByVal + public static native MatVecProdExpression prod(@Const @ByRef MatrixBase a, + @Const @ByRef VectorBase b); + + @ByVal + public static native SrMatDnMatProdExpression prod(@Const @ByRef CompressedMatrix spMx, + @Const @ByRef MatrixBase dMx); + @ByVal + @Name("prod") + public static native DenseColumnMatrix prodCm(@Const @ByRef MatrixBase a, + @Const @ByRef MatrixBase b); + @ByVal + @Name("prod") + public static native DenseRowMatrix prodRm(@Const @ByRef MatrixBase a, + @Const @ByRef MatrixBase b); + + @ByVal + @Name("prod") + public static native DenseRowMatrix prodRm(@Const @ByRef CompressedMatrix spMx, + @Const @ByRef MatrixBase dMx); + + +// @ByVal +// public static native MatrixProdExpression prod(@Const @ByRef DenseRowMatrix a, +// @Const @ByRef DenseRowMatrix b); +// +// @ByVal +// public static native MatrixProdExpression prod(@Const @ByRef DenseRowMatrix a, +// @Const @ByRef DenseColumnMatrix b); +// +// @ByVal +// public static native MatrixProdExpression prod(@Const @ByRef DenseColumnMatrix a, +// @Const @ByRef DenseRowMatrix b); +// +// @ByVal +// public static native MatrixProdExpression prod(@Const @ByRef DenseColumnMatrix a, +// @Const @ByRef DenseColumnMatrix b); + + +} http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/java/org/apache/mahout/viennacl/opencl/javacpp/MatrixTransExpression.scala ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/java/org/apache/mahout/viennacl/opencl/javacpp/MatrixTransExpression.scala b/viennacl/scala-2.11/src/main/java/org/apache/mahout/viennacl/opencl/javacpp/MatrixTransExpression.scala new file mode 100644 index 0000000..115af05 --- /dev/null +++ b/viennacl/scala-2.11/src/main/java/org/apache/mahout/viennacl/opencl/javacpp/MatrixTransExpression.scala @@ -0,0 +1,34 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.mahout.viennacl.opencl.javacpp; + +import org.bytedeco.javacpp.Pointer +import org.bytedeco.javacpp.annotation.{Name, Namespace, Platform, Properties} + + +@Properties(inherit = Array(classOf[Context]), + value = Array(new Platform( + include = Array("matrix.hpp"), + library = "jniViennaCL") + )) +@Namespace("viennacl") +@Name(Array("matrix_expression<const viennacl::matrix_base<double>, " + + "const viennacl::matrix_base<double>, " + + "viennacl::op_trans>")) +class MatrixTransExpression extends Pointer { + +} http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/GPUMMul.scala ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/GPUMMul.scala b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/GPUMMul.scala new file mode 100644 index 0000000..8c6a3a9 --- /dev/null +++ b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/GPUMMul.scala @@ -0,0 +1,444 @@ +/* + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ + +package org.apache.mahout.viennacl.opencl + +import org.apache.mahout.logging._ +import org.apache.mahout.math +import org.apache.mahout.math._ +import org.apache.mahout.math.backend.incore.MMulSolver +import org.apache.mahout.math.flavor.{BackEnum, TraversingStructureEnum} +import org.apache.mahout.math.function.Functions +import org.apache.mahout.math.scalabindings.RLikeOps._ +import org.apache.mahout.math.scalabindings._ +import org.apache.mahout.viennacl.opencl.javacpp.Functions._ +import org.apache.mahout.viennacl.opencl.javacpp.LinalgFunctions._ +import org.apache.mahout.viennacl.opencl.javacpp.{CompressedMatrix, Context, DenseRowMatrix} + +import scala.collection.JavaConversions._ +object GPUMMul extends MMBinaryFunc { + + private implicit val log = getLog(GPUMMul.getClass) + + override def apply(a: Matrix, b: Matrix, r: Option[Matrix]): Matrix = { + + require(a.ncol == b.nrow, "Incompatible matrix sizes in matrix multiplication.") + + val (af, bf) = (a.getFlavor, b.getFlavor) + val backs = (af.getBacking, bf.getBacking) + val sd = (af.getStructure, math.scalabindings.densityAnalysis(a), bf.getStructure, densityAnalysis(b)) + + + try { + + val alg: MMulAlg = backs match { + + // Both operands are jvm memory backs. + case (BackEnum.JVMMEM, BackEnum.JVMMEM) â + + sd match { + + // Multiplication cases by a diagonal matrix. + case (TraversingStructureEnum.VECTORBACKED, _, TraversingStructureEnum.COLWISE, _) + if a.isInstanceOf[DiagonalMatrix] â jvmDiagCW + case (TraversingStructureEnum.VECTORBACKED, _, TraversingStructureEnum.SPARSECOLWISE, _) + if a.isInstanceOf[DiagonalMatrix] â jvmDiagCW + case (TraversingStructureEnum.VECTORBACKED, _, TraversingStructureEnum.ROWWISE, _) + if a.isInstanceOf[DiagonalMatrix] â jvmDiagRW + case (TraversingStructureEnum.VECTORBACKED, _, TraversingStructureEnum.SPARSEROWWISE, _) + if a.isInstanceOf[DiagonalMatrix] â jvmDiagRW + + case (TraversingStructureEnum.COLWISE, _, TraversingStructureEnum.VECTORBACKED, _) + if b.isInstanceOf[DiagonalMatrix] â jvmCWDiag + case (TraversingStructureEnum.SPARSECOLWISE, _, TraversingStructureEnum.VECTORBACKED, _) + if b.isInstanceOf[DiagonalMatrix] â jvmCWDiag + case (TraversingStructureEnum.ROWWISE, _, TraversingStructureEnum.VECTORBACKED, _) + if b.isInstanceOf[DiagonalMatrix] â jvmRWDiag + case (TraversingStructureEnum.SPARSEROWWISE, _, TraversingStructureEnum.VECTORBACKED, _) + if b.isInstanceOf[DiagonalMatrix] â jvmRWDiag + + // Dense-dense cases + case (TraversingStructureEnum.ROWWISE, true, TraversingStructureEnum.COLWISE, true) if a eq b.t â gpuDRWAAt + case (TraversingStructureEnum.ROWWISE, true, TraversingStructureEnum.COLWISE, true) if a.t eq b â gpuDRWAAt + case (TraversingStructureEnum.ROWWISE, true, TraversingStructureEnum.COLWISE, true) â gpuRWCW + case (TraversingStructureEnum.ROWWISE, true, TraversingStructureEnum.ROWWISE, true) â jvmRWRW + case (TraversingStructureEnum.COLWISE, true, TraversingStructureEnum.COLWISE, true) â jvmCWCW + case (TraversingStructureEnum.COLWISE, true, TraversingStructureEnum.ROWWISE, true) if a eq b.t â jvmDCWAAt + case (TraversingStructureEnum.COLWISE, true, TraversingStructureEnum.ROWWISE, true) if a.t eq b â jvmDCWAAt + case (TraversingStructureEnum.COLWISE, true, TraversingStructureEnum.ROWWISE, true) â jvmCWRW + + // Sparse row matrix x sparse row matrix (array of vectors) + case (TraversingStructureEnum.ROWWISE, false, TraversingStructureEnum.ROWWISE, false) â gpuSparseRWRW + case (TraversingStructureEnum.ROWWISE, false, TraversingStructureEnum.COLWISE, false) â jvmSparseRWCW + case (TraversingStructureEnum.COLWISE, false, TraversingStructureEnum.ROWWISE, false) â jvmSparseCWRW + case (TraversingStructureEnum.COLWISE, false, TraversingStructureEnum.COLWISE, false) â jvmSparseCWCW + + // Sparse matrix x sparse matrix (hashtable of vectors) + case (TraversingStructureEnum.SPARSEROWWISE, false, TraversingStructureEnum.SPARSEROWWISE, false) â + gpuSparseRowRWRW + case (TraversingStructureEnum.SPARSEROWWISE, false, TraversingStructureEnum.SPARSECOLWISE, false) â + jvmSparseRowRWCW + case (TraversingStructureEnum.SPARSECOLWISE, false, TraversingStructureEnum.SPARSEROWWISE, false) â + jvmSparseRowCWRW + case (TraversingStructureEnum.SPARSECOLWISE, false, TraversingStructureEnum.SPARSECOLWISE, false) â + jvmSparseRowCWCW + + // Sparse matrix x non-like + case (TraversingStructureEnum.SPARSEROWWISE, false, TraversingStructureEnum.ROWWISE, _) â gpuSparseRowRWRW + case (TraversingStructureEnum.SPARSEROWWISE, false, TraversingStructureEnum.COLWISE, _) â jvmSparseRowRWCW + case (TraversingStructureEnum.SPARSECOLWISE, false, TraversingStructureEnum.ROWWISE, _) â jvmSparseRowCWRW + case (TraversingStructureEnum.SPARSECOLWISE, false, TraversingStructureEnum.COLWISE, _) â jvmSparseCWCW + case (TraversingStructureEnum.ROWWISE, _, TraversingStructureEnum.SPARSEROWWISE, false) â gpuSparseRWRW + case (TraversingStructureEnum.ROWWISE, _, TraversingStructureEnum.SPARSECOLWISE, false) â jvmSparseRWCW + case (TraversingStructureEnum.COLWISE, _, TraversingStructureEnum.SPARSEROWWISE, false) â jvmSparseCWRW + case (TraversingStructureEnum.COLWISE, _, TraversingStructureEnum.SPARSECOLWISE, false) â jvmSparseRowCWCW + + // Everything else including at least one sparse LHS or RHS argument + case (TraversingStructureEnum.ROWWISE, false, TraversingStructureEnum.ROWWISE, _) â gpuSparseRWRW + case (TraversingStructureEnum.ROWWISE, false, TraversingStructureEnum.COLWISE, _) â jvmSparseRWCW + case (TraversingStructureEnum.COLWISE, false, TraversingStructureEnum.ROWWISE, _) â jvmSparseCWRW + case (TraversingStructureEnum.COLWISE, false, TraversingStructureEnum.COLWISE, _) â jvmSparseCWCW2flips + + // Sparse methods are only effective if the first argument is sparse, so we need to do a swap. + case (_, _, _, false) â (a, b, r) â apply(b.t, a.t, r.map { + _.t + }).t + + // Default jvm-jvm case. + // for some reason a SrarseRowMatrix DRM %*% SrarseRowMatrix DRM was dumping off to here + case _ â gpuRWCW + } + } + + alg(a, b, r) + } catch { + // TODO FASTHACK: Revert to JVM if there is an exception. + // E.g., java.lang.nullPointerException if more openCL contexts + // have been created than number of GPU cards. + // Better option wuold be to fall back to OpenCL first. + case ex: Exception => + log.info(ex.getMessage + "falling back to JVM MMUL") + return MMul(a, b, r) + } + } + + type MMulAlg = MMBinaryFunc + + @inline + private def gpuRWCW(a: Matrix, b: Matrix, r: Option[Matrix] = None): Matrix = { + log.info("Using gpuRWCW method") + + val hasElementsA = a.zSum() > 0.0 + val hasElementsB = b.zSum() > 0.0 + + // A has a sparse matrix structure of unknown size. We do not want to + // simply convert it to a Dense Matrix which may result in an OOM error. + + // If it is empty use JVM MMul, since we can not convert it to a VCL CSR Matrix. + if (!hasElementsA) { + log.warn("Matrix a has zero elements can not convert to CSR") + return MMul(a, b, r) + } + + // CSR matrices are efficient up to 50% non-zero + if(b.getFlavor.isDense) { + var ms = System.currentTimeMillis() + val oclCtx = new Context(Context.OPENCL_MEMORY) + val oclA = toVclCmpMatrixAlt(a, oclCtx) + val oclB = toVclDenseRM(b, oclCtx) + val oclC = new DenseRowMatrix(prod(oclA, oclB)) + val mxC = fromVclDenseRM(oclC) + ms = System.currentTimeMillis() - ms + log.debug(s"ViennaCL/OpenCL multiplication time: $ms ms.") + + oclA.close() + oclB.close() + oclC.close() + + mxC + } else { + // Fall back to JVM based MMul if either matrix is sparse and empty + if (!hasElementsA || !hasElementsB) { + log.warn("Matrix a or b has zero elements can not convert to CSR") + return MMul(a, b, r) + } + + var ms = System.currentTimeMillis() + val oclCtx = new Context(Context.OPENCL_MEMORY) + val oclA = toVclCmpMatrixAlt(a, oclCtx) + val oclB = toVclCmpMatrixAlt(b, oclCtx) + val oclC = new CompressedMatrix(prod(oclA, oclB)) + val mxC = fromVclCompressedMatrix(oclC) + ms = System.currentTimeMillis() - ms + log.debug(s"ViennaCL/OpenCL multiplication time: $ms ms.") + + oclA.close() + oclB.close() + oclC.close() + + mxC + } + } + + + @inline + private def jvmRWRW(a: Matrix, b: Matrix, r: Option[Matrix] = None): Matrix = { + log.info("Using jvmRWRW method") + // A bit hackish: currently, this relies a bit on the fact that like produces RW(?) + val bclone = b.like(b.ncol, b.nrow).t + for (brow â b) bclone(brow.index(), ::) := brow + + require(bclone.getFlavor.getStructure == TraversingStructureEnum.COLWISE || bclone.getFlavor.getStructure == + TraversingStructureEnum.SPARSECOLWISE, "COL wise conversion assumption of RHS is wrong, do over this code.") + + gpuRWCW(a, bclone, r) + } + + private def jvmCWCW(a: Matrix, b: Matrix, r: Option[Matrix] = None): Matrix = { + log.info("Using jvmCWCW method") + jvmRWRW(b.t, a.t, r.map(_.t)).t + } + + private def jvmCWRW(a: Matrix, b: Matrix, r: Option[Matrix] = None): Matrix = { + log.info("Using jvmCWRW method") + // This is a primary contender with Outer Prod sum algo. + // Here, we force-reorient both matrices and run RWCW. + // A bit hackish: currently, this relies a bit on the fact that clone always produces RW(?) + val aclone = a.cloned + + require(aclone.getFlavor.getStructure == TraversingStructureEnum.ROWWISE || aclone.getFlavor.getStructure == + TraversingStructureEnum.SPARSEROWWISE, "Row wise conversion assumption of RHS is wrong, do over this code.") + + jvmRWRW(aclone, b, r) + } + + // left is Sparse right is any + private def gpuSparseRWRW(a: Matrix, b: Matrix, r: Option[Matrix] = None): Matrix = { + log.info("Using gpuSparseRWRW method") + val mxR = r.getOrElse(b.like(a.nrow, b.ncol)) + + + /* This is very close to the algorithm from SparseMatrix.times + for (arow â a; ael â arow.nonZeroes) + mxR(arow.index(), ::).assign(b(ael.index, ::), Functions.plusMult(ael)) + mxR + + Make sure that the matrix is not empty. VCL {{compressed_matrix}}s must + have nnz > 0 + N.B. This method is horribly inefficent. However there is a difference between + getNumNonDefaultElements() and getNumNonZeroElements() which we do not always + have access to. We created MAHOUT-1882 for this. + */ + + val hasElementsA = a.zSum() > 0.0 + val hasElementsB = b.zSum() > 0.0 + + // A has a sparse matrix structure of unknown size. We do not want to + // simply convert it to a Dense Matrix which may result in an OOM error. + // If it is empty use JVM MMul, since we can not convert it to a VCL CSR Matrix. + if (!hasElementsA) { + log.warn("Matrix a has zero elements can not convert to CSR") + return MMul(a, b, r) + } + + // CSR matrices are efficient up to 50% non-zero + if(b.getFlavor.isDense) { + var ms = System.currentTimeMillis() + val oclCtx = new Context(Context.OPENCL_MEMORY) + val oclA = toVclCmpMatrixAlt(a, oclCtx) + val oclB = toVclDenseRM(b, oclCtx) + val oclC = new DenseRowMatrix(prod(oclA, oclB)) + val mxC = fromVclDenseRM(oclC) + ms = System.currentTimeMillis() - ms + log.debug(s"ViennaCL/OpenCL multiplication time: $ms ms.") + + oclA.close() + oclB.close() + oclC.close() + + mxC + } else { + // Fall back to JVM based MMul if either matrix is sparse and empty + if (!hasElementsA || !hasElementsB) { + log.warn("Matrix a or b has zero elements can not convert to CSR") + return MMul(a, b, r) + } + + var ms = System.currentTimeMillis() + val oclCtx = new Context(Context.OPENCL_MEMORY) + val oclA = toVclCmpMatrixAlt(a, oclCtx) + val oclB = toVclCmpMatrixAlt(b, oclCtx) + val oclC = new CompressedMatrix(prod(oclA, oclB)) + val mxC = fromVclCompressedMatrix(oclC) + ms = System.currentTimeMillis() - ms + log.debug(s"ViennaCL/OpenCL multiplication time: $ms ms.") + + oclA.close() + oclB.close() + oclC.close() + + mxC + } + + } + + // Sparse %*% dense + private def gpuSparseRowRWRW(a: Matrix, b: Matrix, r: Option[Matrix] = None): Matrix = { + log.info("Using gpuSparseRowRWRW method") + val hasElementsA = a.zSum() > 0 + + // A has a sparse matrix structure of unknown size. We do not want to + // simply convert it to a Dense Matrix which may result in an OOM error. + // If it is empty fall back to JVM MMul, since we can not convert it + // to a VCL CSR Matrix. + if (!hasElementsA) { + log.warn("Matrix a has zero elements can not convert to CSR") + return MMul(a, b, r) + } + + var ms = System.currentTimeMillis() + val oclCtx = new Context(Context.OPENCL_MEMORY) + val oclA = toVclCmpMatrixAlt(a, oclCtx) + val oclB = toVclDenseRM(b, oclCtx) + val oclC = new DenseRowMatrix(prod(oclA, oclB)) + val mxC = fromVclDenseRM(oclC) + ms = System.currentTimeMillis() - ms + log.debug(s"ViennaCL/OpenCL multiplication time: $ms ms.") + + oclA.close() + oclB.close() + oclC.close() + + mxC + } + + private def jvmSparseRowCWCW(a: Matrix, b: Matrix, r: Option[Matrix] = None) = + gpuSparseRowRWRW(b.t, a.t, r.map(_.t)).t + + private def jvmSparseRowCWCW2flips(a: Matrix, b: Matrix, r: Option[Matrix] = None) = + gpuSparseRowRWRW(a cloned, b cloned, r) + + private def jvmSparseRowRWCW(a: Matrix, b: Matrix, r: Option[Matrix]) = + gpuSparseRowRWRW(a, b cloned, r) + + + private def jvmSparseRowCWRW(a: Matrix, b: Matrix, r: Option[Matrix]) = + gpuSparseRowRWRW(a cloned, b, r) + + private def jvmSparseRWCW(a: Matrix, b: Matrix, r: Option[Matrix] = None) = + gpuSparseRWRW(a, b.cloned, r) + + private def jvmSparseCWRW(a: Matrix, b: Matrix, r: Option[Matrix] = None) = + gpuSparseRWRW(a cloned, b, r) + + private def jvmSparseCWCW(a: Matrix, b: Matrix, r: Option[Matrix] = None) = + gpuSparseRWRW(b.t, a.t, r.map(_.t)).t + + private def jvmSparseCWCW2flips(a: Matrix, b: Matrix, r: Option[Matrix] = None) = + gpuSparseRWRW(a cloned, b cloned, r) + + private def jvmDiagRW(diagm:Matrix, b:Matrix, r:Option[Matrix] = None):Matrix = { + log.info("Using jvmDiagRW method") + val mxR = r.getOrElse(b.like(diagm.nrow, b.ncol)) + + for (del â diagm.diagv.nonZeroes()) + mxR(del.index, ::).assign(b(del.index, ::), Functions.plusMult(del)) + + mxR + } + + private def jvmDiagCW(diagm: Matrix, b: Matrix, r: Option[Matrix] = None): Matrix = { + log.info("Using jvmDiagCW method") + val mxR = r.getOrElse(b.like(diagm.nrow, b.ncol)) + for (bcol â b.t) mxR(::, bcol.index()) := bcol * diagm.diagv + mxR + } + + private def jvmCWDiag(a: Matrix, diagm: Matrix, r: Option[Matrix] = None) = + jvmDiagRW(diagm, a.t, r.map {_.t}).t + + private def jvmRWDiag(a: Matrix, diagm: Matrix, r: Option[Matrix] = None) = + jvmDiagCW(diagm, a.t, r.map {_.t}).t + + + /** Dense column-wise AA' */ + private def jvmDCWAAt(a:Matrix, b:Matrix, r:Option[Matrix] = None) = { + // a.t must be equiv. to b. Cloning must rewrite to row-wise. + gpuDRWAAt(a.cloned,null,r) + } + + /** Dense Row-wise AA' */ + // we probably will not want to use this for the actual release unless A is cached already + // but adding for testing purposes. + private def gpuDRWAAt(a:Matrix, b:Matrix, r:Option[Matrix] = None) = { + // a.t must be equiv to b. + log.info("Executing on gpu") + log.debug("AAt computation detected; passing off to GPU") + + // Check dimensions if result is supplied. + require(r.forall(mxR â mxR.nrow == a.nrow && mxR.ncol == a.nrow)) + + val mxR = r.getOrElse(a.like(a.nrow, a.nrow)) + + var ms = System.currentTimeMillis() + val oclCtx = new Context(Context.OPENCL_MEMORY) + val oclA = toVclDenseRM(src = a, oclCtx) + val oclAt = new DenseRowMatrix(trans(oclA)) + val oclC = new DenseRowMatrix(prod(oclA, oclAt)) + + val mxC = fromVclDenseRM(oclC) + ms = System.currentTimeMillis() - ms + debug(s"ViennaCL/OpenCL multiplication time: $ms ms.") + + oclA.close() + oclAt.close() + oclC.close() + + mxC + + } + + private def jvmOuterProdSum(a: Matrix, b: Matrix, r: Option[Matrix] = None): Matrix = { + log.info("Using jvmOuterProdSum method") + // Need to check whether this is already laid out for outer product computation, which may be faster than + // reorienting both matrices. + val (m, n) = (a.nrow, b.ncol) + + // Prefer col-wise result iff a is dense and b is sparse. In all other cases default to row-wise. + val preferColWiseR = a.getFlavor.isDense && !b.getFlavor.isDense + + val mxR = r.getOrElse { + (a.getFlavor.isDense, preferColWiseR) match { + case (false, false) â b.like(m, n) + case (false, true) â b.like(n, m).t + case (true, false) â a.like(m, n) + case (true, true) â a.like(n, m).t + } + } + + // Loop outer products + if (preferColWiseR) { + // B is sparse and A is not, so we need to iterate over b values and update R columns with += + // one at a time. + for ((acol, brow) â a.t.zip(b); bel â brow.nonZeroes) mxR(::, bel.index()) += bel * acol + } else { + for ((acol, brow) â a.t.zip(b); ael â acol.nonZeroes()) mxR(ael.index(), ::) += ael * brow + } + + mxR + } +} http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/CompressedMatrix.scala ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/CompressedMatrix.scala b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/CompressedMatrix.scala new file mode 100644 index 0000000..3fcf9ee --- /dev/null +++ b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/CompressedMatrix.scala @@ -0,0 +1,135 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.mahout.viennacl.opencl.javacpp + +import java.nio._ + +import org.bytedeco.javacpp._ +import org.bytedeco.javacpp.annotation._ + +import scala.collection.mutable.ArrayBuffer + + +@Properties(inherit = Array(classOf[Context]), + value = Array(new Platform( + include = Array("compressed_matrix.hpp"), + library="jniViennaCL" + ))) +@Name(Array("viennacl::compressed_matrix<double>")) +final class CompressedMatrix(defaultCtr: Boolean = true) extends Pointer { + + protected val ptrs = new ArrayBuffer[Pointer]() + + // call this after set or better TODO: yet wrap set() in a public method that will call this + def registerPointersForDeallocation(p:Pointer): Unit = { + ptrs += p + } + + override def deallocate(deallocate: Boolean): Unit = { + super.deallocate(deallocate) + ptrs.foreach(_.close()) + } + + if (defaultCtr) allocate() + + def this(nrow: Int, ncol: Int) { + this(false) + allocate(nrow, ncol, new Context) + } + + def this(nrow: Int, ncol: Int, ctx: Context) { + this(false) + allocate(nrow, ncol, ctx) + } + + def this(nrow: Int, ncol: Int, nonzeros: Int) { + this(false) + allocate(nrow, ncol, nonzeros, new Context) + } + + def this(nrow: Int, ncol: Int, nonzeros: Int, ctx: Context) { + this(false) + allocate(nrow, ncol, nonzeros, ctx) + } + + def this(pe: ProdExpression) { + this(false) + allocate(pe) + } + + @native protected def allocate() + + @native protected def allocate(nrow: Int, ncol: Int, nonzeros: Int, @ByVal ctx: Context) + + @native protected def allocate(nrow: Int, ncol: Int, @ByVal ctx: Context) + + @native protected def allocate(@Const @ByRef pe: ProdExpression) + +// @native protected def allocate(db: DoubleBuffer) +// +// @native protected def allocate(ib: IntBuffer) + + // Warning: apparently there are differences in bit interpretation between OpenCL and everything + // else for unsigned int type. So, for OpenCL backend, rowJumper and colIndices have to be packed + // with reference to that cl_uint type that Vienna-CL defines. + @native def set(@Cast(Array("const void*")) rowJumper: IntBuffer, + @Cast(Array("const void*")) colIndices: IntBuffer, + @Const elements: DoubleBuffer, + nrow: Int, + ncol: Int, + nonzeros: Int + ) + + /** With javacpp pointers. */ + @native def set(@Cast(Array("const void*")) rowJumper: IntPointer, + @Cast(Array("const void*")) colIndices: IntPointer, + @Const elements: DoublePointer, + nrow: Int, + ncol: Int, + nonzeros: Int + ) + + @Name(Array("operator=")) + @native def :=(@Const @ByRef pe: ProdExpression) + + @native def generate_row_block_information() + + /** getters for the compressed_matrix size */ + //const vcl_size_t & size1() const { return rows_; } + @native def size1: Int + //const vcl_size_t & size2() const { return cols_; } + @native def size2: Int + //const vcl_size_t & nnz() const { return nonzeros_; } + @native def nnz: Int + //const vcl_size_t & blocks1() const { return row_block_num_; } + // @native def blocks1: Int + + /** getters for the compressed_matrix buffers */ + //const handle_type & handle1() const { return row_buffer_; } + @native @Const @ByRef def handle1: MemHandle + //const handle_type & handle2() const { return col_buffer_; } + @native @Const @ByRef def handle2: MemHandle + //const handle_type & handle3() const { return row_blocks_; } + @native @Const @ByRef def handle3: MemHandle + //const handle_type & handle() const { return elements_; } + @native @Const @ByRef def handle: MemHandle + +} + +object CompressedMatrix { + Context.loadLib() +} http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/Context.scala ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/Context.scala b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/Context.scala new file mode 100644 index 0000000..770f87f --- /dev/null +++ b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/Context.scala @@ -0,0 +1,73 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.mahout.viennacl.opencl.javacpp + +import org.bytedeco.javacpp.{Loader, Pointer} +import org.bytedeco.javacpp.annotation._ + +/** + * This assumes viennacl 1.7.1 is installed, which in ubuntu Xenial defaults to + * /usr/include/viennacl, and is installed via + * {{{ + * sudo apt-get install libviennacl-dev + * }}} + * + * @param mtype + */ +@Properties(Array( + new Platform( + includepath = Array("/usr/include/viennacl"), + include = Array("matrix.hpp", "compressed_matrix.hpp"), + define = Array("VIENNACL_WITH_OPENCL", "VIENNACL_WITH_OPENMP"), + compiler = Array("fastfpu","viennacl"), + link = Array("OpenCL"), + library = "jniViennaCL" + ))) +@Namespace("viennacl") +@Name(Array("context")) +final class Context(mtype: Int = Context.MEMORY_NOT_INITIALIZED) extends Pointer { + + import Context._ + + if (mtype == MEMORY_NOT_INITIALIZED) + allocate() + else + allocate(mtype) + + @native protected def allocate() + + @native protected def allocate(@Cast(Array("viennacl::memory_types")) mtype: Int) + + @Name(Array("memory_type")) + @Cast(Array("int")) + @native def memoryType: Int + +} + +object Context { + + def loadLib() = Loader.load(classOf[Context]) + + loadLib() + + /* Memory types. Ported from VCL header files. */ + val MEMORY_NOT_INITIALIZED = 0 + val MAIN_MEMORY = 1 + val OPENCL_MEMORY = 2 + val CUDA_MEMORY = 3 + +} http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/DenseColumnMatrix.scala ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/DenseColumnMatrix.scala b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/DenseColumnMatrix.scala new file mode 100644 index 0000000..f12b5c7 --- /dev/null +++ b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/DenseColumnMatrix.scala @@ -0,0 +1,95 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.mahout.viennacl.opencl.javacpp + +import org.bytedeco.javacpp.{DoublePointer, Pointer} +import org.bytedeco.javacpp.annotation._ + +/** + * ViennaCL dense matrix, column-major. This is an exact duplication of [[DenseRowMatrix]], and + * is only different in the materialized C++ template name. Unfortunately I so far have not figured + * out how to handle it with. + * + * Also, the [[Platform.library]] does not get inherited for some reason, and we really want to + * collect all class mappings in the same one libjni.so, so we have to repeat this `library` defi- + * nition in every mapped class in this package. (One .so per package convention). + */ +@Properties(inherit = Array(classOf[Context]), + value = Array(new Platform ( + include=Array("matrix.hpp"), + library="jniViennaCL" + ))) +@Name(Array("viennacl::matrix<double,viennacl::column_major>")) +final class DenseColumnMatrix(initDefault:Boolean = true) extends MatrixBase { + + def this(nrow: Int, ncol: Int) { + this(false) + allocate(nrow, ncol, new Context()) + } + + def this(nrow: Int, ncol: Int, ctx: Context) { + this(false) + allocate(nrow, ncol, ctx) + } + + def this(data: DoublePointer, nrow: Int, ncol: Int ) { + this(false) + allocate(data, new Context(Context.MAIN_MEMORY).memoryType, nrow, ncol) + // We save it to deallocate it ad deallocation time. + ptrs += data + } + + def this(data: DoublePointer, nrow: Int, ncol: Int, ctx: Context) { + this(false) + allocate(data, ctx.memoryType, nrow, ncol) + // We save it to deallocate it ad deallocation time. + ptrs += data + } + + def this(me: MatMatProdExpression) { + this(false) + allocate(me) + } + + def this(me: MatrixTransExpression) { + this(false) + allocate(me) + } + + + if (initDefault) allocate() + + @native protected def allocate() + + @native protected def allocate(nrow: Int, ncol: Int, @ByVal ctx: Context) + + @native protected def allocate(data: DoublePointer, + @Cast(Array("viennacl::memory_types")) + memType: Int, + nrow: Int, + ncol: Int + ) + + @native protected def allocate(@Const @ByRef me: MatMatProdExpression) + + @native protected def allocate(@Const @ByRef me: MatrixTransExpression) + +} + +object DenseColumnMatrix { + Context.loadLib() +} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/DenseRowMatrix.scala ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/DenseRowMatrix.scala b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/DenseRowMatrix.scala new file mode 100644 index 0000000..7014cb2 --- /dev/null +++ b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/DenseRowMatrix.scala @@ -0,0 +1,98 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.mahout.viennacl.opencl.javacpp + +import org.bytedeco.javacpp.{DoublePointer, Pointer, annotation} +import org.bytedeco.javacpp.annotation._ + +import scala.collection.mutable.ArrayBuffer + +/** + * ViennaCL dense matrix, row-major. + */ +@Properties(inherit = Array(classOf[Context]), + value = Array(new Platform( + library = "jniViennaCL" + ))) +@Name(Array("viennacl::matrix<double,viennacl::row_major>")) +class DenseRowMatrix(initDefault: Boolean = true) extends MatrixBase { + + def this(nrow: Int, ncol: Int) { + this(false) + allocate(nrow, ncol, new Context()) + } + + def this(nrow: Int, ncol: Int, ctx: Context) { + this(false) + allocate(nrow, ncol, ctx) + } + + def this(data: DoublePointer, nrow: Int, ncol: Int) { + this(false) + allocate(data, new Context(Context.MAIN_MEMORY).memoryType, nrow, ncol) + // We save it to deallocate it ad deallocation time. + ptrs += data + } + + def this(data: DoublePointer, nrow: Int, ncol: Int, ctx: Context) { + this(false) + allocate(data, ctx.memoryType, nrow, ncol) + // We save it to deallocate it ad deallocation time. + ptrs += data + } + + def this(me: MatMatProdExpression) { + this(false) + allocate(me) + } + + def this(me: MatrixTransExpression) { + this(false) + allocate(me) + } + + // TODO: getting compilation errors here + def this(sd: SrMatDnMatProdExpression) { + this(false) + allocate(sd) + } + + if (initDefault) allocate() + + @native protected def allocate() + + @native protected def allocate(nrow: Int, ncol: Int, @ByVal ctx: Context) + + @native protected def allocate(data: DoublePointer, + @Cast(Array("viennacl::memory_types")) + memType: Int, + nrow: Int, + ncol: Int + ) + + @native protected def allocate(@Const @ByRef me: MatMatProdExpression) + + @native protected def allocate(@Const @ByRef me: MatrixTransExpression) + + @native protected def allocate(@Const @ByRef me: SrMatDnMatProdExpression) + +} + + +object DenseRowMatrix { + Context.loadLib() +} \ No newline at end of file http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MatMatProdExpression.scala ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MatMatProdExpression.scala b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MatMatProdExpression.scala new file mode 100644 index 0000000..c88aee5 --- /dev/null +++ b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MatMatProdExpression.scala @@ -0,0 +1,33 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.mahout.viennacl.opencl.javacpp + +import org.bytedeco.javacpp.Pointer +import org.bytedeco.javacpp.annotation.{Name, Namespace, Platform, Properties} + + +@Properties(inherit = Array(classOf[Context]), + value = Array(new Platform( + library = "jniViennaCL") + )) +@Namespace("viennacl") +@Name(Array("matrix_expression<const viennacl::matrix_base<double>, " + + "const viennacl::matrix_base<double>, " + + "viennacl::op_mat_mat_prod>")) +class MatMatProdExpression extends Pointer { + +} http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MatVecProdExpression.scala ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MatVecProdExpression.scala b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MatVecProdExpression.scala new file mode 100644 index 0000000..111cbd3 --- /dev/null +++ b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MatVecProdExpression.scala @@ -0,0 +1,33 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.mahout.viennacl.opencl.javacpp + +import org.bytedeco.javacpp.Pointer +import org.bytedeco.javacpp.annotation.{Name, Namespace, Platform, Properties} + + +@Properties(inherit = Array(classOf[Context]), + value = Array(new Platform( + library = "jniViennaCL") + )) +@Namespace("viennacl") +@Name(Array("vector_expression<const viennacl::matrix_base<double>, " + + "const viennacl::vector_base<double>, " + + "viennacl::op_prod>")) +class MatVecProdExpression extends Pointer { + +} http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MatrixBase.scala ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MatrixBase.scala b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MatrixBase.scala new file mode 100644 index 0000000..6cc1f9f --- /dev/null +++ b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MatrixBase.scala @@ -0,0 +1,75 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.mahout.viennacl.opencl.javacpp + +import org.bytedeco.javacpp.Pointer +import org.bytedeco.javacpp.annotation._ + +import scala.collection.mutable.ArrayBuffer + + +@Properties(inherit = Array(classOf[Context]), + value = Array(new Platform( + library = "jniViennaCL" + ))) +@Name(Array("viennacl::matrix_base<double>")) +class MatrixBase extends Pointer { + + protected val ptrs = new ArrayBuffer[Pointer]() + + override def deallocate(deallocate: Boolean): Unit = { + super.deallocate(deallocate) + ptrs.foreach(_.close()) + } + + @Name(Array("operator=")) + @native def :=(@Const @ByRef src: DenseRowMatrix) + + @Name(Array("operator=")) + @native def :=(@Const @ByRef src: DenseColumnMatrix) + + @Name(Array("size1")) + @native + def nrow: Int + + @Name(Array("size2")) + @native + def ncol: Int + + @Name(Array("row_major")) + @native + def isRowMajor: Boolean + + @Name(Array("internal_size1")) + @native + def internalnrow: Int + + @Name(Array("internal_size2")) + @native + def internalncol: Int + + @Name(Array("memory_domain")) + @native + def memoryDomain: Int + + @Name(Array("switch_memory_context")) + @native + def switchMemoryContext(@ByRef ctx: Context) + + + +} http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MemHandle.scala ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MemHandle.scala b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MemHandle.scala new file mode 100644 index 0000000..73807ac --- /dev/null +++ b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/MemHandle.scala @@ -0,0 +1,48 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.mahout.viennacl.opencl.javacpp + +import org.bytedeco.javacpp.{Loader, Pointer} +import org.bytedeco.javacpp.annotation._ + + +@Properties(inherit = Array(classOf[Context]), + value = Array(new Platform( + library = "jniViennaCL") + )) +@Namespace("viennacl::backend") +@Name(Array("mem_handle")) +class MemHandle extends Pointer { + + allocate() + + @native def allocate() +} + +object MemHandle { + + def loadLib() = Loader.load(classOf[MemHandle]) + + loadLib() + + /* Memory types. Ported from VCL header files. */ + val MEMORY_NOT_INITIALIZED = 0 + val MAIN_MEMORY = 1 + val OPENCL_MEMORY = 2 + val CUDA_MEMORY = 3 + +} http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/ProdExpression.scala ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/ProdExpression.scala b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/ProdExpression.scala new file mode 100644 index 0000000..7ee42b8 --- /dev/null +++ b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/ProdExpression.scala @@ -0,0 +1,33 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.mahout.viennacl.opencl.javacpp + +import org.bytedeco.javacpp.Pointer +import org.bytedeco.javacpp.annotation.{Name, Namespace, Platform, Properties} + + +@Properties(inherit = Array(classOf[Context]), + value = Array(new Platform( + library = "jniViennaCL") + )) +@Namespace("viennacl") +@Name(Array("matrix_expression<const viennacl::compressed_matrix<double>, " + + "const viennacl::compressed_matrix<double>, " + + "viennacl::op_prod>")) +class ProdExpression extends Pointer { + +} http://git-wip-us.apache.org/repos/asf/mahout/blob/9a4f9d36/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/SrMatDnMatProdExpression.scala ---------------------------------------------------------------------- diff --git a/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/SrMatDnMatProdExpression.scala b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/SrMatDnMatProdExpression.scala new file mode 100644 index 0000000..24d2c7b --- /dev/null +++ b/viennacl/scala-2.11/src/main/scala/org/apache/mahout/viennacl/opencl/javacpp/SrMatDnMatProdExpression.scala @@ -0,0 +1,33 @@ +/** + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed with + * this work for additional information regarding copyright ownership. + * The ASF licenses this file to You under the Apache License, Version 2.0 + * (the "License"); you may not use this file except in compliance with + * the License. You may obtain a copy of the License at + * + * http://www.apache.org/licenses/LICENSE-2.0 + * + * Unless required by applicable law or agreed to in writing, software + * distributed under the License is distributed on an "AS IS" BASIS, + * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * See the License for the specific language governing permissions and + * limitations under the License. + */ +package org.apache.mahout.viennacl.opencl.javacpp + +import org.bytedeco.javacpp.Pointer +import org.bytedeco.javacpp.annotation.{Name, Namespace, Platform, Properties} + + +@Properties(inherit = Array(classOf[Context]), + value = Array(new Platform( + library = "jniViennaCL") + )) +@Namespace("viennacl") +@Name(Array("matrix_expression<const viennacl::compressed_matrix<double>, " + + "const viennacl::matrix_base<double>, " + + "viennacl::op_prod>")) +class SrMatDnMatProdExpression extends Pointer { + +}
