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

more cleaning

Browse Source
This commit is contained in:
Philippe Tillet
2019-10-11 19:29:24 -04:00
parent 323c90e431
commit ee3803b577
4 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
lib/codegen/analysis/layout.cc
View File

@@ -222,6 +222,7 @@ layout_shared_t::layout_shared_t(const layout_t *arg,
size_t _id, size_t _id,
analysis::align* align): layout_t(SHARED, _axes, _shapes, values, _id, align) { analysis::align* align): layout_t(SHARED, _axes, _shapes, values, _id, align) {
size = 0;
if(arg->type == SCANLINE) if(arg->type == SCANLINE)
order = arg->order; order = arg->order;

2
lib/codegen/analysis/liveness.cc
View File

@@ -154,7 +154,7 @@ void liveness::run(ir::module &mod) {
// connected components // connected components
for(auto &x: layouts_->get_all()) { for(auto &x: layouts_->get_all()) {
layout_t* layout = x.second; layout_t*& layout = x.second;
if(layout->type != SHARED) if(layout->type != SHARED)
continue; continue;
for(ir::value *v: layout->values) for(ir::value *v: layout->values)

1
lib/driver/module.cc
View File

@@ -241,7 +241,6 @@ std::string cu_module::compile_llvm_module(std::unique_ptr<llvm::Module> module,
cu_module::cu_module(driver::context * context, std::unique_ptr<llvm::Module> ll_module): cu_module(context, compile_llvm_module(std::move(ll_module), context->device())) { } cu_module::cu_module(driver::context * context, std::unique_ptr<llvm::Module> ll_module): cu_module(context, compile_llvm_module(std::move(ll_module), context->device())) { }
cu_module::cu_module(driver::context * context, std::string const & source) : module(context, CUmodule(), true), source_(source){ cu_module::cu_module(driver::context * context, std::string const & source) : module(context, CUmodule(), true), source_(source){
std::cout << source << std::endl;
cu_context::context_switcher ctx(*context); cu_context::context_switcher ctx(*context);
// JIT compile source-code // JIT compile source-code
CUjit_option opt[] = {CU_JIT_ERROR_LOG_BUFFER_SIZE_BYTES, CU_JIT_ERROR_LOG_BUFFER}; CUjit_option opt[] = {CU_JIT_ERROR_LOG_BUFFER_SIZE_BYTES, CU_JIT_ERROR_LOG_BUFFER};

1
lib/runtime/function.cc
View File

@@ -245,6 +245,7 @@ std::unique_ptr<driver::module> function::make_bin(ir::module &module, driver::c
align.run(module); align.run(module);
axes.run(module); axes.run(module);
layouts.run(module); layouts.run(module);
liveness.run(module);
// ir::print(module, std::cout); // ir::print(module, std::cout);
selection.run(module, *llvm); selection.run(module, *llvm);
// return binary // return binary