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

added tensorflow code generator

Browse Source
This commit is contained in:
Philippe Tillet
2019-08-15 15:59:53 -07:00
parent 38a8b0ab19
commit 3ece461ce2
4 changed files with 174 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
examples/cpp/dot.cpp
View File

@@ -154,6 +154,9 @@ perf_t do_bench(drv::stream* stream, bool AT, bool BT, int32_t M, int32_t N, int
stream->synchronize();
// run
rt::function function(src(AT, BT, ty, ty, ty, 8, 8));
std::cout << function.make_tensorflow_src({2}, "(M + #TM - 1)/#TM, (N + #TN - 1)/#TN, 1") << std::endl;
exit(EXIT_FAILURE);
auto ceil = [](size_t x, size_t y) { return (x + y - 1) / y; };
auto grid = [&](const rt::params_t& x) { return rt::grid_t{ceil(M, x.at("TM")), ceil(N, x.at("TN")), 1}; };
@@ -202,7 +205,7 @@ int main() {
// shapes to benchmark
std::vector<config_t> configs = {
// {false, false, 8192, 512, 512},
{false, true, 8192, 8192, 8192}
{false, true, 128, 128, 128}
// {false, true, 128, 128, 128},
// {false, false, 128, 128, 128},
// {true, false, 128, 128, 128},