gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added GEMM benchmark

Browse Source
This commit is contained in:
Philippe Tillet
2015-01-25 18:19:19 -05:00
parent 4a9e16fefd
commit 6d7f103ab1
7 changed files with 132 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
python/pyatidlas/src/_atidlas.cpp
View File

@@ -77,17 +77,17 @@ bp::tuple get_shape(atd::array const & x)
// x.reshape(size1, size2);
//}
boost::python::dict create_queues(atd::cl::queues_t queues)
{
boost::python::dict dictionary;
for (atd::cl::queues_t::iterator it = queues.begin(); it != queues.end(); ++it) {
bp::list list;
for (atd::cl::queues_t::mapped_type::iterator itt = it->second.begin(); itt != it->second.end(); ++itt)
list.append(*itt);
dictionary[it->first] = list;
}
return dictionary;
}
//boost::python::dict create_queues(atd::cl::queues_t queues)
//{
// boost::python::dict dictionary;
// for (atd::cl::queues_t::iterator it = queues.begin(); it != queues.end(); ++it) {
// bp::list list;
// for (atd::cl::queues_t::mapped_type::iterator itt = it->second.begin(); itt != it->second.end(); ++itt)
// list.append(*itt);
// dictionary[it->first] = list;
// }
// return dictionary;
//}
template<class T>
struct datatype : public atd::value_scalar
@@ -198,7 +198,7 @@ namespace detail
}
std::vector<atd::cl::CommandQueue> & get_queue(atd::cl::Context const & ctx)
{ return atd::cl::queues[ctx]; }
{ return atd::cl::get_queues(ctx); }
atd::numeric_type extract_dtype(bp::object const & odtype)
{