void TestKernel::edgeTypes()
{
GraphDocumentPtr document = GraphDocument::create();
document->edgeTypes().first()->setId(1);
EdgeTypePtr typeB = EdgeType::create(document);
typeB->setId(2);
NodePtr nodeA = Node::create(document);
NodePtr nodeB = Node::create(document);
EdgePtr edge = Edge::create(nodeA, nodeB);
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
script = "Document.edges()[0].type;";
result = kernel.execute(document, script);
QCOMPARE(result.toString().toInt(), 1);
script = "Document.edges()[0].type = 2;";
kernel.execute(document, script);
QCOMPARE(edge->type()->id(), 2);
script = "Document.edges()[0].type;";
result = kernel.execute(document, script);
QCOMPARE(result.toString().toInt(), 2);
// cleanup
document->destroy();
}
void TestKernel::distance()
{
GraphDocumentPtr document = GraphDocument::create();
NodePtr nodeA = Node::create(document);
NodePtr nodeB = Node::create(document);
NodePtr nodeC = Node::create(document);
EdgePtr edgeAB = Edge::create(nodeA, nodeB);
EdgePtr edgeBC = Edge::create(nodeB, nodeC);
document->edgeTypes().first()->addDynamicProperty("dist");
document->edgeTypes().first()->setDirection(EdgeType::Bidirectional);
edgeAB->setDynamicProperty("dist", "1");
edgeBC->setDynamicProperty("dist", "1");
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
script = "Document.nodes()[0].distance(\"dist\", Document.nodes())[2];";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(2));
script = "Document.nodes()[2].distance(\"dist\", Document.nodes())[0];";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(2));
// cleanup
document->destroy();
}
void TestKernel::nodeDynamicProperties()
{
GraphDocumentPtr document = GraphDocument::create();
NodeTypePtr type = document->nodeTypes().first();
NodePtr node = Node::create(document);
type->addDynamicProperty("propertyA");
type->addDynamicProperty("propertyB");
type->addDynamicProperty("propertyC");
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
// property read-access from script
node->setDynamicProperty("propertyA", "1");
script = "Document.nodes()[0].propertyA;";
result = kernel.execute(document, script);
QCOMPARE(result.toString().toInt(), 1);
// property local write/read-access in script
script = "Document.nodes()[0].propertyB = 2; Document.nodes()[0].propertyB";
kernel.execute(document, script);
QCOMPARE(node->dynamicProperty("propertyB").toInt(), 2);
// property write-access from script
script = "Document.nodes()[0].propertyC = 3";
kernel.execute(document, script);
QCOMPARE(node->dynamicProperty("propertyC").toInt(), 3);
// cleanup
document->destroy();
}
void TestKernel::edgeProperties()
{
GraphDocumentPtr document = GraphDocument::create();
document->edgeTypes().first()->setDirection(EdgeType::Unidirectional);
NodePtr nodeA = Node::create(document);
NodePtr nodeB = Node::create(document);
EdgePtr edge = Edge::create(nodeA, nodeB);
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
script = "Document.nodes()[0].edges()[0].from().id;";
result = kernel.execute(document, script);
QCOMPARE(result.toString().toInt(), nodeA->id());
script = "Document.nodes()[0].edges()[0].to().id;";
result = kernel.execute(document, script);
QCOMPARE(result.toString().toInt(), nodeB->id());
script = "Document.nodes()[0].edges()[0].directed();";
result = kernel.execute(document, script);
QCOMPARE(result.toBool(), true);
// cleanup
document->destroy();
}
void TestKernel::nodeProperties()
{
GraphDocumentPtr document = GraphDocument::create();
document->edgeTypes().first()->setDirection(EdgeType::Unidirectional);
NodePtr node = Node::create(document);
node->setId(1);
node->setX(20);
node->setY(30);
node->setColor("#ff0000");
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
script = "Document.nodes()[0].id;";
result = kernel.execute(document, script);
QCOMPARE(result.toString().toInt(), node->id());
script = "Document.nodes()[0].x;";
result = kernel.execute(document, script);
QCOMPARE(result.toString().toDouble(), qreal(node->x()));
script = "Document.nodes()[0].y;";
result = kernel.execute(document, script);
QCOMPARE(result.toString().toDouble(), qreal(node->y()));
script = "Document.nodes()[0].color;";
result = kernel.execute(document, script);
QCOMPARE(result.toString(), QString("#ff0000"));
// cleanup
document->destroy();
}
void TestKernel::edgeAccessMethodsWithTypes()
{
GraphDocumentPtr document = GraphDocument::create();
document->edgeTypes().first()->setDirection(EdgeType::Bidirectional);
NodePtr nodeA = Node::create(document);
NodePtr nodeB = Node::create(document);
NodePtr nodeC = Node::create(document);
EdgePtr edgeAB = Edge::create(nodeA, nodeB);
EdgePtr edgeBC = Edge::create(nodeB, nodeC);
// test edge type
EdgeTypePtr edgeTypeB = EdgeType::create(document);
edgeTypeB->setDirection(EdgeType::Unidirectional);
edgeBC->setType(edgeTypeB);
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
script = QString("Document.nodes()[1].edges(%1).length;").arg(edgeTypeB->id());
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = QString("Document.nodes()[1].inEdges(%1).length;").arg(edgeTypeB->id());
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(0));
script = QString("Document.nodes()[1].outEdges(%1).length;").arg(edgeTypeB->id());
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
// cleanup
document->destroy();
}
void TestKernel::edgeAccessMethods()
{
GraphDocumentPtr document = GraphDocument::create();
document->edgeTypes().first()->setDirection(EdgeType::Unidirectional);
NodePtr nodeA = Node::create(document);
NodePtr nodeB = Node::create(document);
EdgePtr edge = Edge::create(nodeA, nodeB);
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
script = "Document.nodes()[0].edges().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[0].inEdges().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(0));
script = "Document.nodes()[0].outEdges().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
// cleanup
document->destroy();
}
void TestKernel::automaticScriptObjectPropertyGeneration()
{
GraphDocumentPtr document = GraphDocument::create();
NodePtr nodeA = Node::create(document);
NodePtr nodeB = Node::create(document);
EdgePtr edge = Edge::create(nodeA, nodeB);
Kernel kernel;
QString script;
QScriptValue result;
// For edges/nodes we can assign arbitrary dynamic properties during
// script execution. However, they exist only during execution and are removed
// at the end of the execution.
script = "Document.nodes()[0].nonRegProp=1; Document.nodes()[0].nonRegProp;";
result = kernel.execute(document, script);
QCOMPARE(result.toString().toInt(), 1);
QCOMPARE(nodeA->dynamicProperty("nonRegProp").toInt(), 0);
script = "Document.edges()[0].nonRegProp=1; Document.edges()[0].nonRegProp;";
result = kernel.execute(document, script);
QCOMPARE(result.toString().toInt(), 1);
QCOMPARE(edge->dynamicProperty("nonRegProp").toInt(), 0);
// cleanup
document->destroy();
}
void TestKernel::graphObjectAccessWithTypes()
{
GraphDocumentPtr document = GraphDocument::create();
NodeTypePtr nodeTypeB = NodeType::create(document);
EdgeTypePtr edgeTypeB = EdgeType::create(document);
NodePtr nodeA = Node::create(document);
NodePtr nodeB = Node::create(document);
NodePtr nodeC = Node::create(document);
EdgePtr edgeAB = Edge::create(nodeA, nodeB);
EdgePtr edgeBC = Edge::create(nodeB, nodeC);
// setup types
nodeA->setType(nodeTypeB);
edgeAB->setType(edgeTypeB);
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
script = QString("Document.nodes(%1).length;").arg(nodeTypeB->id());
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = QString("Document.edges(%1).length;").arg(edgeTypeB->id());
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
// cleanup
document->destroy();
}
void MatrixCSR<scalar,GPU>::test1(const Vector<scalar, GPU>& in, Vector<scalar, GPU>& out, cl_mem devTexVec)
{
cl_int err;
cl_event ocl_event;
size_t localWorkSize[1];
size_t globalWorkSize[1];
// localWorkSize[0] = 256;
// globalWorkSize[0] = roundUp(localWorkSize[0], in.get_csize());
localWorkSize[0] = 128;
globalWorkSize[0] =( size_t( nrow / localWorkSize[0] ) + 1 ) * localWorkSize[0];
Kernel k = OpenCL::kernels["s_kernel_test"];
k.set_int(0, this->nrow);
k.set_buffer(1, this->mat.row_offset);
k.set_buffer(2, this->mat.col);
k.set_buffer(3, this->mat.val);
k.set_buffer(4, devTexVec);
k.set_buffer(5, out.get_data());
k.execute(localWorkSize[0], globalWorkSize[0], &ocl_event);
err = clWaitForEvents( 1, &ocl_event );
//printf("csr test kernel wait event: %s\n", OpenCL::getError (err).toStdString().c_str());
err = clReleaseEvent( ocl_event );
}
void TestKernel::engineSetup()
{
GraphDocumentPtr document;
NodePtr nodeA, nodeB;
EdgePtr edge;
QVERIFY(GraphDocument::objects() == 0);
QVERIFY(Node::objects() == 0);
QVERIFY(Edge::objects() == 0);
// test destroy graph document
document = GraphDocument::create();
nodeA = Node::create(document);
nodeB = Node::create(document);
edge = Edge::create(nodeA, nodeB);
// create kernel
QString script = "return true;";
Kernel kernel;
QScriptValue result = kernel.execute(document, script);
QCOMPARE(result.toBool(), true);
document->destroy();
document.reset();
nodeA.reset();
nodeB.reset();
edge.reset();
QCOMPARE(Edge::objects(), uint(0));
QCOMPARE(Node::objects(), uint(0));
QCOMPARE(GraphDocument::objects(), uint(0));
}
void TestKernel::graphObjectAccess()
{
GraphDocumentPtr document = GraphDocument::create();
NodePtr nodeA = Node::create(document);
NodePtr nodeB = Node::create(document);
EdgePtr edge = Edge::create(nodeA, nodeB);
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
script = "Document.nodes().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(2));
script = "Document.edges().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
// cleanup
document->destroy();
}
void TestKernel::createNode()
{
GraphDocumentPtr document = GraphDocument::create();
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
script = "Document.createNode(0, 0);";
result = kernel.execute(document, script);
QCOMPARE(document->nodes().count(), 1);
// cleanup
document->destroy();
}
void TestKernel::createEdge()
{
GraphDocumentPtr document = GraphDocument::create();
NodePtr nodeA = Node::create(document);
NodePtr nodeB = Node::create(document);
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
script = "Document.createEdge(Document.nodes()[0], Document.nodes()[1]);";
result = kernel.execute(document, script);
QCOMPARE(document->edges().count(), 1);
// cleanup
document->destroy();
}
void TestKernel::nodeAccessMethods()
{
GraphDocumentPtr document = GraphDocument::create();
NodePtr node = Node::create(document);
node->setId(42);
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
script = "Document.node(42).id;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(42));
// cleanup
document->destroy();
}
void MatrixCSR<scalar,GPU>::test2(const Vector<scalar, GPU>& in, Vector<scalar, GPU>& out, cl_mem devTexVec)
{
assert(in.get_csize() >= 0);
assert(out.get_size()>= 0);
assert(in.get_csize() == this->get_ncol());
assert(out.get_size()== this->get_nrow());
//const int nnz_per_row = get_nnz() / get_nrow();
//printf("nnz_per_row: %d\n", nnz_per_row);
cl_int err;
cl_event ocl_event;
size_t localWorkSize[1];
size_t globalWorkSize[1];
//TODO: Tune up those parameters!
const int WARP_SIZE = 32;
localWorkSize[0] = WARP_SIZE * 8;
globalWorkSize[0] = localWorkSize[0] * 128;
Kernel k;
if (sizeof(scalar) == 4)
k = OpenCL::kernels["s_kernel_test2"];
else
{
fprintf(stdout, "WRONG KERNEL!!! DOUBLE IS NOT HERE YET"); exit(-100);
}
k.set_int(0, this->nrow);
k.set_buffer(1, this->mat.row_offset);
k.set_buffer(2, this->mat.col);
k.set_buffer(3, this->mat.val);
k.set_buffer(4, devTexVec);
k.set_buffer(5, out.get_data());
k.set_local(6, (localWorkSize[0]+16)*sizeof(scalar) );
k.execute(localWorkSize[0], globalWorkSize[0], &ocl_event);
err = clWaitForEvents( 1, &ocl_event );
//printf("csr kernel wait event: %s\n", OpenCL::getError (err).toStdString().c_str());
err = clReleaseEvent( ocl_event );
}
void MatrixCSR<scalar,GPU>::multiply(const Vector<scalar, GPU>& in, Vector<scalar, GPU>& out, bool use_scalar)
{
assert(in.get_csize() >= 0);
assert(out.get_size()>= 0);
assert(in.get_csize() == this->get_ncol());
assert(out.get_size()== this->get_nrow());
//const int nnz_per_row = get_nnz() / get_nrow();
//printf("nnz_per_row: %d\n", nnz_per_row);
cl_event ocl_event;
size_t localWorkSize[1];
size_t globalWorkSize[1];
//USE IT TO DECIDE SCALAR OR VECTOR
//const scalar avg_nnz_per_row = ((scalar)this->get_nnz())/this->get_nrow();
if (use_scalar)
{
//qDebug() << "CRS Multiply: USING SCALAR KERNEL\n";
localWorkSize[0] = 128;
globalWorkSize[0] = ( size_t( nrow / localWorkSize[0] ) + 1 ) * localWorkSize[0];
Kernel k;
if (sizeof(scalar) == 4)
k = OpenCL::kernels["s_kernel_csr_spmv_scalar"];
else
k = OpenCL::kernels["d_kernel_csr_spmv_scalar"];
k.set_int(0, this->nrow);
k.set_buffer(1, this->mat.row_offset);
k.set_buffer(2, this->mat.col);
k.set_buffer(3, this->mat.val);
k.set_buffer(4, in.get_cdata());
k.set_buffer(5, out.get_data());
k.execute(localWorkSize[0], globalWorkSize[0], &ocl_event);
cl_int err = clWaitForEvents( 1, &ocl_event );
//printf("csr scalar kernel wait event: %s\n", OpenCL::getError (err).toStdString().c_str());
err = clReleaseEvent( ocl_event );
}
else
{
//TODO: Tune up those parameters!
const int WARP_SIZE = 32;
localWorkSize[0] = WARP_SIZE * 8;
globalWorkSize[0] = localWorkSize[0] * 128;
Kernel k;
if (sizeof(scalar) == 4)
k = OpenCL::kernels["s_kernel_csr_spmv_vector"];
else
k = OpenCL::kernels["d_kernel_csr_spmv_vector"];
k.set_int(0, this->nrow);
k.set_buffer(1, this->mat.row_offset);
k.set_buffer(2, this->mat.col);
k.set_buffer(3, this->mat.val);
k.set_buffer(4, in.get_cdata());
k.set_buffer(5, out.get_data());
k.set_local(6, (localWorkSize[0]+16)*sizeof(scalar) );
k.execute(localWorkSize[0], globalWorkSize[0], &ocl_event);
cl_int err = clWaitForEvents( 1, &ocl_event );
//printf("csr kernel wait event: %s\n", OpenCL::getError (err).toStdString().c_str());
err = clReleaseEvent( ocl_event );
}
}
void TestKernel::neighborships()
{
GraphDocumentPtr document = GraphDocument::create();
document->edgeTypes().first()->setDirection(EdgeType::Unidirectional);
NodePtr nodeA = Node::create(document);
NodePtr nodeB = Node::create(document);
EdgePtr edge = Edge::create(nodeA, nodeB);
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
// test with unidirectional edge
script = "Document.nodes()[0].neighbors().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[1].neighbors().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[0].successors().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[1].successors().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(0));
script = "Document.nodes()[0].predecessors().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(0));
script = "Document.nodes()[1].predecessors().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
// test with bidirectional edge
document->edgeTypes().first()->setDirection(EdgeType::Bidirectional);
script = "Document.nodes()[0].neighbors().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[1].neighbors().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[0].successors().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[1].successors().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[0].predecessors().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[1].predecessors().length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
// cleanup
document->destroy();
}
void TestKernel::neighborshipsWithTypes()
{
// this test is the same as withouth types but we add an backedge of another type that
// changes the results if types are not respected
GraphDocumentPtr document = GraphDocument::create();
EdgeTypePtr typeA = document->edgeTypes().first();
EdgeTypePtr typeB = EdgeType::create(document);
typeA->setDirection(EdgeType::Unidirectional);
typeA->setId(1);
typeB->setDirection(EdgeType::Unidirectional);
typeB->setId(2);
NodePtr nodeA = Node::create(document);
NodePtr nodeB = Node::create(document);
EdgePtr edgeAB = Edge::create(nodeA, nodeB);
EdgePtr edgeBA = Edge::create(nodeB, nodeA);
edgeBA->setType(typeB);
// test nodes
Kernel kernel;
QString script;
QScriptValue result;
// test with unidirectional edge
script = "Document.nodes()[0].neighbors(1).length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[1].neighbors(1).length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[0].successors(1).length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[1].successors(1).length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(0));
script = "Document.nodes()[0].predecessors(1).length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(0));
script = "Document.nodes()[1].predecessors(1).length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
// test with bidirectional edge
document->edgeTypes().first()->setDirection(EdgeType::Bidirectional);
script = "Document.nodes()[0].neighbors(1).length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[1].neighbors(1).length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[0].successors(1).length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[1].successors(1).length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[0].predecessors(1).length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
script = "Document.nodes()[1].predecessors(1).length;";
result = kernel.execute(document, script);
QCOMPARE(result.toInteger(), qreal(1));
// cleanup
document->destroy();
}
