github/AMD-SHARK-Studio
1
1
Fork 0
mirror of https://github.com/nod-ai/AMD-SHARK-Studio.git synced 2026-04-03 03:00:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

wip for dynamic indexes

Browse Source
This commit is contained in:
Elias
2022-03-10 18:45:09 +00:00
parent 5223ff69fe
commit e5174ca097
2 changed files with 16 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
dSHARK/minimal-http-server-form-post-file.c
View File

@@ -222,9 +222,10 @@ void sigint_handler(int sig)
interrupted = 1;
}
void *run_module_thread()
void* run_module_thread(void *i)
{
int x = run_module();
int index = *(int *)i;
int x = run_module(index);
return NULL;
}
@@ -232,9 +233,11 @@ int main(int argc, const char **argv)
{
pthread_t t1, t2;
int i1, i2;
int32_t ci0 = 0;
int32_t ci1 = 1;
printf("Before Thread\n");
i1 = pthread_create(&t1, NULL, run_module_thread, NULL);
i2 = pthread_create(&t2, NULL, run_module_thread, NULL);
i1 = pthread_create(&t1, NULL, run_module_thread, (void *)0);
i2 = pthread_create(&t2, NULL, run_module_thread, (void *)1);
pthread_join(t1, NULL);
pthread_join(t2, NULL);
printf("After Thread\n");

16
dSHARK/run-module.c
View File

@@ -19,15 +19,17 @@
#include "iree/modules/hal/module.h"
#include "iree/vm/api.h"
#include "iree/vm/bytecode_module.h"
#include "iree/base/internal/flags.h"
#include "iree/hal/cuda/registration/driver_module.h"
#include "iree/base/tracing.h"
#include "iree/hal/cuda/api.h"
#include "simple_embedding_test_bytecode_module_cuda_c.h"
iree_status_t create_sample_device(iree_allocator_t host_allocator,
iree_hal_device_t** out_device) {
iree_hal_device_t** out_device, int index) {
// Only register the CUDA HAL driver.
IREE_RETURN_IF_ERROR(
iree_hal_cuda_driver_module_register(iree_hal_driver_registry_default()));
iree_hal_cuda_driver_module_register(iree_hal_driver_registry_default(), index));
// Create the HAL driver from the name.
iree_hal_driver_t* driver = NULL;
@@ -52,7 +54,7 @@ const iree_const_byte_span_t load_bytecode_module_data() {
module_file_toc->size);
}
iree_status_t Run() {
iree_status_t Run(int index) {
// TODO(benvanik): move to instance-based registration.
IREE_RETURN_IF_ERROR(iree_hal_module_register_types());
@@ -61,7 +63,7 @@ iree_status_t Run() {
iree_vm_instance_create(iree_allocator_system(), &instance));
iree_hal_device_t* device = NULL;
IREE_RETURN_IF_ERROR(create_sample_device(iree_allocator_system(), &device),
IREE_RETURN_IF_ERROR(create_sample_device(iree_allocator_system(), &device, index),
"create device");
iree_vm_module_t* hal_module = NULL;
IREE_RETURN_IF_ERROR(
@@ -181,8 +183,8 @@ iree_status_t Run() {
return iree_ok_status();
}
int run_module() {
const iree_status_t result = Run();
int run_module(int index) {
const iree_status_t result = Run(index);
int ret = (int)iree_status_code(result);
if (!iree_status_is_ok(result)) {
iree_status_fprint(stderr, result);