blob: 20f4333061dcadcccb457d39830e6ab7d1be6909 [file] [log] [blame]
/*
* 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.sysds.runtime.instructions.gpu;
import org.apache.commons.lang3.tuple.Pair;
import org.apache.sysds.common.Types;
import org.apache.sysds.hops.codegen.cplan.CNodeCell;
import org.apache.sysds.runtime.codegen.CodegenUtils;
import org.apache.sysds.runtime.codegen.SpoofCellwise;
import org.apache.sysds.runtime.codegen.SpoofOperator;
import org.apache.sysds.runtime.codegen.SpoofCUDA;
import org.apache.sysds.runtime.controlprogram.caching.MatrixObject;
import org.apache.sysds.runtime.controlprogram.context.ExecutionContext;
import org.apache.sysds.runtime.instructions.InstructionUtils;
import org.apache.sysds.runtime.instructions.cp.CPOperand;
import org.apache.sysds.runtime.instructions.cp.ScalarObject;
import org.apache.sysds.runtime.lineage.LineageItem;
import org.apache.sysds.runtime.lineage.LineageItemUtils;
import org.apache.sysds.runtime.lineage.LineageTraceable;
import org.apache.sysds.runtime.instructions.cp.DoubleObject;
import java.util.ArrayList;
public class SpoofCUDAInstruction extends GPUInstruction implements LineageTraceable {
private final SpoofCUDA _op;
private final CPOperand[] _in;
public final CPOperand _out;
private SpoofCUDAInstruction(SpoofOperator op, CPOperand[] in, CPOperand out, String opcode, String istr) {
super(null, opcode, istr);
if(!(op instanceof SpoofCUDA))
throw new RuntimeException("SpoofGPUInstruction needs an operator of type SpoofNativeCUDA!");
_op = (SpoofCUDA) op;
_in = in;
_out = out;
instString = istr;
instOpcode = opcode;
}
public static SpoofCUDAInstruction parseInstruction(String str) {
String[] parts = InstructionUtils.getInstructionPartsWithValueType(str);
ArrayList<CPOperand> inlist = new ArrayList<>();
SpoofCUDA op = CodegenUtils.getNativeOpData(parts[2]);
String opcode = op.getSpoofType();
for( int i=3; i<parts.length-2; i++ )
inlist.add(new CPOperand(parts[i]));
CPOperand out = new CPOperand(parts[parts.length-2]);
return new SpoofCUDAInstruction(op, inlist.toArray(new CPOperand[0]), out, opcode, str);
}
@Override
public void processInstruction(ExecutionContext ec) {
//get input matrices and scalars, incl pinning of matrices
ArrayList<MatrixObject> inputs = new ArrayList<>();
ArrayList<ScalarObject> scalars = new ArrayList<>();
for (CPOperand input : _in) {
if(input.getDataType()== Types.DataType.MATRIX)
inputs.add(ec.getMatrixInputForGPUInstruction(input.getName(), getExtendedOpcode()));
else if(input.getDataType()== Types.DataType.SCALAR) {
//note: even if literal, it might be compiled as scalar placeholder
scalars.add(ec.getScalarInput(input));
}
}
// set the output dimensions to the hop node matrix dimensions
if( _out.getDataType() == Types.DataType.MATRIX) {
long rows = inputs.get(0).getNumRows();
long cols = inputs.get(0).getNumColumns();
if(_op.getSpoofTemplateType().contains("CW"))
if(((CNodeCell)_op.getCNodeTemplate()).getCellType() == SpoofCellwise.CellType.COL_AGG)
rows = 1;
else if(((CNodeCell)_op.getCNodeTemplate()).getCellType() == SpoofCellwise.CellType.ROW_AGG)
cols = 1;
MatrixObject out_obj = ec.getDenseMatrixOutputForGPUInstruction(_out.getName(), rows, cols).getKey();
ec.setMetaData(_out.getName(), out_obj.getNumRows(), out_obj.getNumColumns());
_op.execute(inputs, scalars, out_obj, ec);
ec.releaseMatrixOutputForGPUInstruction(_out.getName());
}
else if (_out.getDataType() == Types.DataType.SCALAR) {
ScalarObject out = new DoubleObject(_op.execute(inputs, scalars, null, ec));
ec.setScalarOutput(_out.getName(), out);
}
for (CPOperand input : _in)
if(input.getDataType()== Types.DataType.MATRIX)
ec.releaseMatrixInputForGPUInstruction(input.getName());
}
@Override
public Pair<String, LineageItem> getLineageItem(ExecutionContext ec) {
return Pair.of(_out.getName(),
new LineageItem(getOpcode(), LineageItemUtils.getLineage(ec, _in)));
}
}