use of org.apache.sysml.runtime.functionobjects.Modulus in project incubator-systemml by apache.
the class LibMatrixCUDA method matrixMatrixOp.
/**
* Utility to launch binary cellwise matrix-matrix operations CUDA kernel
*
* @param gCtx a valid {@link GPUContext}
* @param ec execution context
* @param instName the invoking instruction's name for record {@link Statistics}.
* @param in1 left input matrix
* @param in2 right input matrix
* @param outputName output variable name
* @param isLeftTransposed true if left matrix is transposed
* @param isRightTransposed true if right matrix is transposed
* @param op operator
*/
private static void matrixMatrixOp(ExecutionContext ec, GPUContext gCtx, String instName, MatrixObject in1, MatrixObject in2, String outputName, boolean isLeftTransposed, boolean isRightTransposed, BinaryOperator op) {
if (ec.getGPUContext(0) != gCtx)
throw new DMLRuntimeException("GPU : Invalid internal state, the GPUContext set with the ExecutionContext is not the same used to run this LibMatrixCUDA function");
boolean isEmpty1 = isSparseAndEmpty(gCtx, in1);
boolean isEmpty2 = isSparseAndEmpty(gCtx, in2);
int rlenA = toInt(in1.getNumRows());
int rlenB = toInt(in2.getNumRows());
int clenA = toInt(in1.getNumColumns());
int clenB = toInt(in2.getNumColumns());
int vecStatusA = getVectorStatus(rlenA, clenA).code();
int vecStatusB = getVectorStatus(rlenB, clenB).code();
if (isLeftTransposed || isRightTransposed) {
throw new DMLRuntimeException("Unsupported operator: GPU transposed binary op " + isLeftTransposed + " " + isRightTransposed);
}
long outRLen = Math.max(rlenA, rlenB);
long outCLen = Math.max(clenA, clenB);
if (isEmpty1 && isEmpty2) {
MatrixObject out = ec.allocateGPUMatrixObject(outputName, outRLen, outCLen);
// When both inputs are empty, the output is empty too (except in the case of division)
if (op.fn instanceof Divide || op.fn instanceof IntegerDivide || op.fn instanceof Modulus) {
out.getGPUObject(gCtx).allocateAndFillDense(Double.NaN);
} else if (op.fn instanceof Minus1Multiply) {
out.getGPUObject(gCtx).allocateAndFillDense(1.0);
} else {
out.getGPUObject(gCtx).allocateSparseAndEmpty();
}
} else // Check for M1 * M2 when M1 is empty; if M2 is a vector then fallback to general case
if (isEmpty1 && clenB != 1 && rlenB != 1) {
// C = empty_in1 op in2 ==> becomes ==> C = 0.0 op in2
matrixScalarArithmetic(ec, gCtx, instName, in2, outputName, isRightTransposed, new LeftScalarOperator(op.fn, 0.0));
} else // Check for M1 * M2 when M2 is empty; if M1 is a vector then fallback to general case
if (isEmpty2 && clenA != 1 && rlenA != 1) {
// C = in1 op empty_in2 ==> becomes ==> C = in1 op 0.0
matrixScalarArithmetic(ec, gCtx, instName, in1, outputName, isLeftTransposed, new RightScalarOperator(op.fn, 0.0));
} else {
// TODO: FIXME: Implement sparse binCellSparseOp kernel
Pointer A = getDensePointer(gCtx, in1, instName);
// TODO: FIXME: Implement sparse binCellSparseOp kernel
Pointer B = getDensePointer(gCtx, in2, instName);
// Allocated the dense output matrix
MatrixObject out = null;
try {
out = getDenseMatrixOutputForGPUInstruction(ec, instName, outputName, outRLen, outCLen);
} catch (DMLRuntimeException e) {
throw new DMLRuntimeException("Incorrect dimensions: dimA:[" + rlenA + "," + clenA + "]" + " dimB:[" + rlenB + "," + clenB + "] out:[" + outRLen + "," + outCLen + "]", e);
}
Pointer C = getDensePointer(gCtx, out, instName);
int maxRlen = Math.max(rlenA, rlenB);
int maxClen = Math.max(clenA, clenB);
matrixMatrixOp(gCtx, instName, A, B, maxRlen, maxClen, vecStatusA, vecStatusB, C, op);
}
}
use of org.apache.sysml.runtime.functionobjects.Modulus in project incubator-systemml by apache.
the class ElementwiseModulusTest method testSparse.
@Test
public void testSparse() {
int rows = 50;
int cols = 50;
TestConfiguration config = availableTestConfigurations.get("SparseTest");
config.addVariable("rows", rows);
config.addVariable("cols", cols);
loadTestConfiguration(config);
double[][] a = getRandomMatrix(rows, cols, -5, 5, 0.05, -1);
double[][] b = getNonZeroRandomMatrix(rows, cols, -20, 20, -1);
double[][] c = new double[rows][cols];
Modulus fnmod = Modulus.getFnObject();
for (int i = 0; i < rows; i++) {
for (int j = 0; j < cols; j++) {
c[i][j] = fnmod.execute(a[i][j], b[i][j]);
}
}
writeInputMatrix("a", a);
writeInputMatrix("b", b);
writeExpectedMatrix("c", c);
runTest();
compareResults();
}
use of org.apache.sysml.runtime.functionobjects.Modulus in project incubator-systemml by apache.
the class ScalarModulusTest method testIntVar.
@Test
public void testIntVar() {
int rows = 10;
int cols = 10;
int divisor = 20;
int dividend = 20;
TestConfiguration config = availableTestConfigurations.get("IntVarTest");
config.addVariable("rows", rows);
config.addVariable("cols", cols);
config.addVariable("vardeclaration", "divisor = " + divisor + "; dividend = " + dividend + ";");
config.addVariable("divisor", "divisor");
config.addVariable("dividend", "dividend");
loadTestConfiguration(config);
double[][] vector = getNonZeroRandomMatrix(rows, 1, 1, 5, -1);
double[][] computedVectorLeft = new double[rows][1];
double[][] computedVectorRight = new double[rows][1];
Modulus fnmod = Modulus.getFnObject();
for (int i = 0; i < rows; i++) {
computedVectorLeft[i][0] = fnmod.execute(vector[i][0], divisor);
computedVectorRight[i][0] = fnmod.execute(dividend, vector[i][0]);
}
writeInputMatrix("vector", vector);
writeExpectedMatrix("vector_left", computedVectorLeft);
writeExpectedMatrix("vector_right", computedVectorRight);
double[][] matrix = getNonZeroRandomMatrix(rows, cols, 1, 5, -1);
double[][] computedMatrixLeft = new double[rows][cols];
double[][] computedMatrixRight = new double[rows][cols];
for (int i = 0; i < rows; i++) {
for (int j = 0; j < cols; j++) {
computedMatrixLeft[i][j] = fnmod.execute(matrix[i][j], divisor);
computedMatrixRight[i][j] = fnmod.execute(dividend, matrix[i][j]);
}
}
writeInputMatrix("matrix", matrix);
writeExpectedMatrix("matrix_left", computedMatrixLeft);
writeExpectedMatrix("matrix_right", computedMatrixRight);
runTest();
compareResults();
}
use of org.apache.sysml.runtime.functionobjects.Modulus in project incubator-systemml by apache.
the class ScalarModulusTest method testDoubleVar.
@Test
public void testDoubleVar() {
int rows = 10;
int cols = 10;
int divisor = 20;
int dividend = 20;
TestConfiguration config = availableTestConfigurations.get("DoubleVarTest");
config.addVariable("rows", rows);
config.addVariable("cols", cols);
config.addVariable("vardeclaration", "divisor = " + divisor + "; dividend = " + dividend + ";");
config.addVariable("divisor", "divisor");
config.addVariable("dividend", "dividend");
loadTestConfiguration(config);
double[][] vector = getNonZeroRandomMatrix(rows, 1, 1, 5, -1);
double[][] computedVectorLeft = new double[rows][1];
double[][] computedVectorRight = new double[rows][1];
Modulus fnmod = Modulus.getFnObject();
for (int i = 0; i < rows; i++) {
computedVectorLeft[i][0] = fnmod.execute(vector[i][0], divisor);
computedVectorRight[i][0] = fnmod.execute(dividend, vector[i][0]);
}
writeInputMatrix("vector", vector);
writeExpectedMatrix("vector_left", computedVectorLeft);
writeExpectedMatrix("vector_right", computedVectorRight);
double[][] matrix = getNonZeroRandomMatrix(rows, cols, 1, 5, -1);
double[][] computedMatrixLeft = new double[rows][cols];
double[][] computedMatrixRight = new double[rows][cols];
for (int i = 0; i < rows; i++) {
for (int j = 0; j < cols; j++) {
computedMatrixLeft[i][j] = fnmod.execute(matrix[i][j], divisor);
computedMatrixRight[i][j] = fnmod.execute(dividend, matrix[i][j]);
}
}
writeInputMatrix("matrix", matrix);
writeExpectedMatrix("matrix_left", computedMatrixLeft);
writeExpectedMatrix("matrix_right", computedMatrixRight);
runTest();
compareResults();
}
use of org.apache.sysml.runtime.functionobjects.Modulus in project incubator-systemml by apache.
the class ScalarModulusTest method testDivisionByZero.
@Test
public void testDivisionByZero() {
int rows = 10;
int cols = 10;
int divisor = 0;
int dividend = 0;
TestConfiguration config = availableTestConfigurations.get("DivisionByZeroTest");
config.addVariable("rows", rows);
config.addVariable("cols", cols);
config.addVariable("vardeclaration", "");
config.addVariable("divisor", divisor);
config.addVariable("dividend", dividend);
loadTestConfiguration(config);
double[][] vector = getRandomMatrix(rows, 1, 1, 5, 0.5, -1);
double[][] computedVectorLeft = new double[rows][1];
double[][] computedVectorRight = new double[rows][1];
Modulus fnmod = Modulus.getFnObject();
for (int i = 0; i < rows; i++) {
computedVectorLeft[i][0] = fnmod.execute(vector[i][0], divisor);
computedVectorRight[i][0] = fnmod.execute(dividend, vector[i][0]);
}
writeInputMatrix("vector", vector);
writeExpectedMatrix("vector_left", computedVectorLeft);
writeExpectedMatrix("vector_right", computedVectorRight);
double[][] matrix = getRandomMatrix(rows, cols, -1, 1, 0.5, -1);
double[][] computedMatrixLeft = new double[rows][cols];
double[][] computedMatrixRight = new double[rows][cols];
for (int i = 0; i < rows; i++) {
for (int j = 0; j < cols; j++) {
computedMatrixLeft[i][j] = fnmod.execute(matrix[i][j], divisor);
computedMatrixRight[i][j] = fnmod.execute(dividend, matrix[i][j]);
}
}
writeInputMatrix("matrix", matrix);
writeExpectedMatrix("matrix_left", computedMatrixLeft);
writeExpectedMatrix("matrix_right", computedMatrixRight);
runTest();
compareResults();
}
Aggregations