diff options
Diffstat (limited to 'modules/remote-kokkos')
-rw-r--r-- | modules/remote-kokkos/.nix/derivation.nix | 30 | ||||
-rw-r--r-- | modules/remote-kokkos/SConstruct | 71 | ||||
-rw-r--r-- | modules/remote-kokkos/c++/SConscript | 38 | ||||
-rw-r--r-- | modules/remote-kokkos/c++/device.hpp | 14 | ||||
-rw-r--r-- | modules/remote-kokkos/c++/remote.hpp | 48 | ||||
-rw-r--r-- | modules/remote-kokkos/c++/rpc.hpp | 47 | ||||
-rw-r--r-- | modules/remote-kokkos/tests/SConscript | 31 |
7 files changed, 279 insertions, 0 deletions
diff --git a/modules/remote-kokkos/.nix/derivation.nix b/modules/remote-kokkos/.nix/derivation.nix new file mode 100644 index 0000000..dc9cd94 --- /dev/null +++ b/modules/remote-kokkos/.nix/derivation.nix @@ -0,0 +1,30 @@ +{ lib +, stdenv +, scons +, clang-tools +, version +, forstio +}: + +let + +in stdenv.mkDerivation { + pname = "forstio-device-hip"; + inherit version; + src = ./..; + + enableParallelBuilding = true; + + nativeBuildInputs = [ + scons + clang-tools + ]; + + buildInputs = [ + forstio.core + forstio.async + forstio.codec + ]; + + outputs = ["out" "dev"]; +} diff --git a/modules/remote-kokkos/SConstruct b/modules/remote-kokkos/SConstruct new file mode 100644 index 0000000..c5b2bb4 --- /dev/null +++ b/modules/remote-kokkos/SConstruct @@ -0,0 +1,71 @@ +#!/usr/bin/env python3 + +import sys +import os +import os.path +import glob +import re + + +if sys.version_info < (3,): + def isbasestring(s): + return isinstance(s,basestring) +else: + def isbasestring(s): + return isinstance(s, (str,bytes)) + +def add_kel_source_files(self, sources, filetype, lib_env=None, shared=False, target_post=""): + + if isbasestring(filetype): + dir_path = self.Dir('.').abspath + filetype = sorted(glob.glob(dir_path+"/"+filetype)) + + for path in filetype: + target_name = re.sub( r'(.*?)(\.cpp|\.c\+\+)', r'\1' + target_post, path ) + if shared: + target_name+='.os' + sources.append( self.SharedObject( target=target_name, source=path ) ) + else: + target_name+='.o' + sources.append( self.StaticObject( target=target_name, source=path ) ) + pass + +def isAbsolutePath(key, dirname, env): + assert os.path.isabs(dirname), "%r must have absolute path syntax" % (key,) + +env_vars = Variables( + args=ARGUMENTS +) + +env_vars.Add('prefix', + help='Installation target location of build results and headers', + default='/usr/local/', + validator=isAbsolutePath +) + +env=Environment(ENV=os.environ, variables=env_vars, CPPPATH=[], + CPPDEFINES=['SAW_UNIX'], + CXXFLAGS=['-std=c++20','-g','-Wall','-Wextra'], + LIBS=[ + 'forstio-core' + 'forstio-codec' + ] +); +env.__class__.add_source_files = add_kel_source_files +env.Tool('compilation_db'); +env.cdb = env.CompilationDatabase('compile_commands.json'); + +env.objects = []; +env.sources = []; +env.headers = []; +env.targets = []; + +Export('env') +SConscript('c++/SConscript') +SConscript('tests/SConscript') + +env.Alias('cdb', env.cdb); +env.Alias('all', [env.targets]); +env.Default('all'); + +env.Alias('install', '$prefix') diff --git a/modules/remote-kokkos/c++/SConscript b/modules/remote-kokkos/c++/SConscript new file mode 100644 index 0000000..b66ce79 --- /dev/null +++ b/modules/remote-kokkos/c++/SConscript @@ -0,0 +1,38 @@ +#!/bin/false + +import os +import os.path +import glob + + +Import('env') + +dir_path = Dir('.').abspath + +# Environment for base library +dev_hip_env = env.Clone(); + +dev_hip_env.sources = sorted(glob.glob(dir_path + "/*.cpp")) +dev_hip_env.headers = sorted(glob.glob(dir_path + "/*.hpp")) + +env.sources += dev_hip_env.sources; +env.headers += dev_hip_env.headers; + +## Shared lib +objects_shared = [] +dev_hip_env.add_source_files(objects_shared, dev_hip_env.sources, shared=True); +env.library_shared = dev_hip_env.SharedLibrary('#build/forstio-device-hip', [objects_shared]); + +## Static lib +objects_static = [] +dev_hip_env.add_source_files(objects_static, dev_hip_env.sources, shared=False); +env.library_static = dev_hip_env.StaticLibrary('#build/forstio-device-hip', [objects_static]); + +# Set Alias +env.Alias('library_device-hip', [env.library_shared, env.library_static]); + +env.targets += ['library_device-hip']; + +# Install +env.Install('$prefix/lib/', [env.library_shared, env.library_static]); +env.Install('$prefix/include/forstio/device/hip/', [dev_hip_env.headers]); diff --git a/modules/remote-kokkos/c++/device.hpp b/modules/remote-kokkos/c++/device.hpp new file mode 100644 index 0000000..6c84fe1 --- /dev/null +++ b/modules/remote-kokkos/c++/device.hpp @@ -0,0 +1,14 @@ +#pragma once + +namespace saw { +namespace dev { +struct Hip {}; +} +template<typename T> +struct backend; + +struct backend<dev::Hip> { +}; + + +} diff --git a/modules/remote-kokkos/c++/remote.hpp b/modules/remote-kokkos/c++/remote.hpp new file mode 100644 index 0000000..549b9ec --- /dev/null +++ b/modules/remote-kokkos/c++/remote.hpp @@ -0,0 +1,48 @@ +#pragma once + +#include <Kokkos_Core.hpp> + +namespace saw { +namespace rmt { +struct Kokkos {}; +} + +template<typename T> +struct remote; + +template<typename T> +struct remote_ctx; + +template<> +class remote_ctx<rmt::Kokkos> { +private: + Kokkos::ScopeGuard guard_; + + SAW_FORBID_COPY(remote_ctx); + SAW_FORBID_MOVE(remote_ctx); +public: + remote_ctx(Kokkos::InitializationSettings init_settings): + guard_{std::move(init_settings)} + {} + + error_or<void> create_remote(){ + return remote<rmt::Kokkos>{*this}; + } +}; + +template<> +struct remote<rmt::Kokkos> { +private: + remote_ctx<rmt::Kokkos>* ctx_; + + SAW_FORBID_COPY(remote); + SAW_FORBID_MOVE(remote); +public: + remote(remote_ctx<rmt::Kokkos>& r_ctx): + ctx_{&r_ctx} + {} + + template<typename Iface> + error +}; +} diff --git a/modules/remote-kokkos/c++/rpc.hpp b/modules/remote-kokkos/c++/rpc.hpp new file mode 100644 index 0000000..5bce553 --- /dev/null +++ b/modules/remote-kokkos/c++/rpc.hpp @@ -0,0 +1,47 @@ +#pragma once + +#include <hip.hpp> + +namespace saw { +namespace rmt { +struct DeviceHip {}; +} + +template<> +class remote<rmt::DeviceHip> { +private: +public: + template<typename Iface> + error_or<rpc_client<rmt::DeviceHip, Iface>> connect(){ + return make_error<err::not_implemented>(); + } +}; + +error_or<remote<rmt::DeviceHip>> create_remote(){ + auto rc = hipInit(0); + + return make_error<err::not_implemented>(); +} + +template<typename Iface> +class rpc_client<rmt::DeviceHip, Iface> { +public: + template<typename... T> + struct request { + std::tuple<id<T>...> ids; + + error_or<data<schema::Tuple<T...>>> wait(); + }; +}; + +template<typename Iface> +class rpc_server<rmt::DeviceHip, Iface> { +private: + /** + * Needs a variant ptr of all possible return types + */ + struct data_storage { + }; + +}; +} diff --git a/modules/remote-kokkos/tests/SConscript b/modules/remote-kokkos/tests/SConscript new file mode 100644 index 0000000..f8ffc92 --- /dev/null +++ b/modules/remote-kokkos/tests/SConscript @@ -0,0 +1,31 @@ +#!/bin/false + +import os +import os.path +import glob + + +Import('env') + +dir_path = Dir('.').abspath + +# Environment for base library +test_cases_env = env.Clone(); + +test_cases_env.Append(LIBS=['forstio-test']); + +test_cases_env.sources = sorted(glob.glob(dir_path + "/*.cpp")) +test_cases_env.headers = sorted(glob.glob(dir_path + "/*.hpp")) + +env.sources += test_cases_env.sources; +env.headers += test_cases_env.headers; + +objects_static = [] +test_cases_env.add_source_files(objects_static, test_cases_env.sources, shared=False); +test_cases_env.program = test_cases_env.Program('#bin/tests', [objects_static, env.library_static]); + +# Set Alias +env.Alias('test', test_cases_env.program); +env.Alias('check', test_cases_env.program); + +env.targets += ['test','check']; |