0
0
Fork 0
mirror of https://github.com/matrix-construct/construct synced 2024-12-25 15:04:10 +01:00
construct/ircd/gpt_pipe_code.cc

304 lines
4.7 KiB
C++

// Tensor Construct
//
// Copyright (C) Matrix Construct Developers, Authors & Contributors
// Copyright (C) 2016-2022 Jason Volk <jason@zemos.net>
//
// Permission to use, copy, modify, and/or distribute this software for any
// purpose with or without fee is hereby granted, provided that the above
// copyright notice and this permission notice is present in all copies. The
// full license for this software is available in the LICENSE file.
decltype(ircd::gpt::pipe::code::default_path)
ircd::gpt::pipe::code::default_path
{
{ "name", "ircd.gpt.pipe.code.path" },
};
decltype(ircd::gpt::pipe::code::cache_path)
ircd::gpt::pipe::code::cache_path
{
{ "name", "ircd.gpt.pipe.code.cache.path" },
};
decltype(ircd::gpt::pipe::code::default_compile_opts)
ircd::gpt::pipe::code::default_compile_opts
{
{ "name", "ircd.gpt.pipe.code.opts.comp" },
{ "default", string_view
{
//" -fident"
//" -fno-builtin"
//" -ffast-math"
//" -O3"
" -cl-no-signed-zeros"
" -cl-finite-math-only"
" -cl-fp32-correctly-rounded-divide-sqrt"
" -cl-single-precision-constant"
" -cl-kernel-arg-info"
}}
};
decltype(ircd::gpt::pipe::code::default_link_opts)
ircd::gpt::pipe::code::default_link_opts
{
{ "name", "ircd.gpt.pipe.code.opts.link" },
{ "default", string_view
{
//" -fident"
//" -fno-builtin"
//" -ffast-math"
//" -O3"
//" -cl-no-signed-zeros"
//" -cl-finite-math-only"
//" -cl-fp32-correctly-rounded-divide-sqrt"
//" -cl-single-precision-constant"
}}
};
//
// code::code
//
ircd::gpt::pipe::code::code()
:cl::code{[]() -> cl::code
{
const auto comp_opts
{
fmt::snstringf
{
4096, "%s -I%s",
string_view{default_compile_opts},
string_view{fs::base::include},
}
};
const string_view link_opts
{
default_link_opts
};
cl::code ret;
if(!ret)
ret = from_cache();
if(!ret)
ret = from_source(comp_opts, link_opts);
if(!ret)
ret = from_bitcode(link_opts);
return ret;
}()}
{
put_cache();
}
ircd::gpt::pipe::code::~code()
noexcept
{
}
bool
ircd::gpt::pipe::code::put_cache()
{
const auto cache_path
{
make_cache_path(fs::path_scratch)
};
if(!cache_path)
return false;
if(fs::is_reg(cache_path))
return false;
set_cache(cache_path);
return true;
}
void
ircd::gpt::pipe::code::set_cache(const string_view &path)
{
const unique_mutable_buffer cache_buf
{
this->bins_size()
};
mutable_buffer _cache_bufs[1]
{
cache_buf
};
const auto cache_bufs
{
this->bin(_cache_bufs)
};
const fs::fd fd
{
path, fs::fd::opts
{
.mode = std::ios::out
},
};
const auto written
{
fs::write(fd, cache_bufs[0])
};
assert(!empty(written));
}
extern const uint8_t
gpt_gpu_r600_barts_bc[],
gpt_gpu_spv[];
extern const uint
gpt_gpu_r600_barts_bc_len,
gpt_gpu_spv_len;
ircd::cl::code
ircd::gpt::pipe::code::from_bitcode(const string_view &link_opts)
{
const const_buffer bitcode
{
reinterpret_cast<const char *>(gpt_gpu_r600_barts_bc),
gpt_gpu_r600_barts_bc_len
//reinterpret_cast<const char *>(gpt_gpu_spv),
//gpt_gpu_spv_len
};
char pbuf[1][48];
log::logf
{
log, log::level::DEBUG,
"bitcode %p %s link_opts:%zu attempting...",
data(bitcode),
pretty(pbuf[0], si(size(bitcode))),
size(link_opts),
};
cl::code ret
{
bitcode
};
ret.link(link_opts);
return ret;
}
ircd::cl::code
ircd::gpt::pipe::code::from_source(const string_view &comp_opts,
const string_view &link_opts)
{
const string_view code_path
{
default_path
};
if(!code_path)
return {};
log::logf
{
log, log::level::DEBUG,
"source code `%s'comp_opts:%zu link_opts:%zu attempting...",
code_path,
size(comp_opts),
size(link_opts),
};
cl::code ret
{
cl::code::path, code_path
};
ret.compile(comp_opts);
ret.link(link_opts);
return ret;
}
ircd::cl::code
ircd::gpt::pipe::code::from_cache()
{
const auto cache_path
{
make_cache_path(fs::path_scratch)
};
if(!cache_path)
return {};
if(!fs::is_reg(cache_path))
return {};
const fs::fd fd
{
cache_path
};
const std::string read
{
fs::read(fd)
};
const const_buffer bin
{
read
};
const vector_view<const const_buffer> bins
(
&bin, 1
);
char pbuf[1][48];
log::logf
{
log, log::level::DEBUG,
"cached nir `%s' %s attempting...",
cache_path,
pretty(pbuf[0], si(size(read))),
};
return cl::code
{
bins
};
}
ircd::string_view
ircd::gpt::pipe::code::make_cache_path(const mutable_buffer &buf)
{
if(!cache_path)
return {};
const string_view &src_path
{
default_path
};
const string_view &src_fname
{
fs::filename(fs::path_scratch, src_path)
};
const string_view &cache_fname
{
fs::extension(fs::name_scratch, src_fname, ".r600_barts.bc")
};
const string_view parts[]
{
string_view{cache_path},
string_view{cache_fname}
};
return fs::path(buf, fs::path_views{parts});
}