Maneesh Gupta
62 řádky
1.7 KiB
C++
62 řádky
1.7 KiB
C++
#include "../include/hip/hcc_detail/code_object_bundle.hpp"
|
|
|
|
#include <hsa/hsa.h>
|
|
|
|
#include <algorithm>
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
using namespace std;
|
|
|
|
hsa_isa_t hip_impl::triple_to_hsa_isa(const std::string& triple) {
|
|
static constexpr const char prefix[] = "hcc-amdgcn--amdhsa-gfx";
|
|
static constexpr size_t prefix_sz = sizeof(prefix) - 1;
|
|
|
|
hsa_isa_t r = {};
|
|
|
|
auto idx = triple.find(prefix);
|
|
|
|
if (idx != string::npos) {
|
|
idx += prefix_sz;
|
|
string tmp = "AMD:AMDGPU";
|
|
while (idx != triple.size()) {
|
|
tmp.push_back(':');
|
|
tmp.push_back(triple[idx++]);
|
|
}
|
|
|
|
hsa_isa_from_name(tmp.c_str(), &r);
|
|
}
|
|
|
|
return r;
|
|
}
|
|
|
|
// DATA - STATICS
|
|
constexpr const char hip_impl::Bundled_code_header::magic_string_[];
|
|
|
|
// CREATORS
|
|
hip_impl::Bundled_code_header::Bundled_code_header(const vector<char>& x)
|
|
: Bundled_code_header{x.cbegin(), x.cend()} {}
|
|
|
|
hip_impl::Bundled_code_header::Bundled_code_header(
|
|
const void* p) { // This is a pretty terrible interface, useful only because
|
|
// hipLoadModuleData is so poorly specified (for no fault of its own).
|
|
if (!p) return;
|
|
|
|
auto ph = static_cast<const Header_*>(p);
|
|
|
|
if (!equal(magic_string_, magic_string_ + magic_string_sz_, ph->bundler_magic_string_)) {
|
|
return;
|
|
}
|
|
|
|
size_t sz = sizeof(Header_) + ph->bundle_cnt_ * sizeof(Bundled_code::Header);
|
|
auto pb = static_cast<const char*>(p) + sizeof(Header_);
|
|
auto n = ph->bundle_cnt_;
|
|
while (n--) {
|
|
sz += reinterpret_cast<const Bundled_code::Header*>(pb)->bundle_sz;
|
|
pb += sizeof(Bundled_code::Header);
|
|
}
|
|
|
|
read(static_cast<const char*>(p), static_cast<const char*>(p) + sz, *this);
|
|
} |