-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
References for 'BlueBrain/nmodl#1311'.
- Loading branch information
GitHub Actions Bot
committed
Jun 11, 2024
1 parent
e07df6d
commit 88191f9
Showing
2 changed files
with
683 additions
and
0 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,354 @@ | ||
/********************************************************* | ||
Model Name : NetReceiveCalls | ||
Filename : NetReceiveCalls.mod | ||
NMODL Version : 7.7.0 | ||
Vectorized : true | ||
Threadsafe : true | ||
Created : DATE | ||
Simulator : CoreNEURON | ||
Backend : C++ (api-compatibility) | ||
NMODL Compiler : VERSION | ||
*********************************************************/ | ||
|
||
#include <math.h> | ||
#include <stdio.h> | ||
#include <stdlib.h> | ||
#include <string.h> | ||
|
||
#include <coreneuron/gpu/nrn_acc_manager.hpp> | ||
#include <coreneuron/mechanism/mech/mod2c_core_thread.hpp> | ||
#include <coreneuron/mechanism/register_mech.hpp> | ||
#include <coreneuron/nrnconf.h> | ||
#include <coreneuron/nrniv/nrniv_decl.h> | ||
#include <coreneuron/sim/multicore.hpp> | ||
#include <coreneuron/sim/scopmath/newton_thread.hpp> | ||
#include <coreneuron/utils/ivocvect.hpp> | ||
#include <coreneuron/utils/nrnoc_aux.hpp> | ||
#include <coreneuron/utils/randoms/nrnran123.h> | ||
|
||
|
||
namespace coreneuron { | ||
#ifndef NRN_PRCELLSTATE | ||
#define NRN_PRCELLSTATE 0 | ||
#endif | ||
|
||
|
||
/** channel information */ | ||
static const char *mechanism_info[] = { | ||
"7.7.0", | ||
"NetReceiveCalls", | ||
0, | ||
"c1", | ||
"c2", | ||
0, | ||
0, | ||
0 | ||
}; | ||
|
||
|
||
/** all global variables */ | ||
struct NetReceiveCalls_Store { | ||
int point_type{}; | ||
int reset{}; | ||
int mech_type{}; | ||
}; | ||
static_assert(std::is_trivially_copy_constructible_v<NetReceiveCalls_Store>); | ||
static_assert(std::is_trivially_move_constructible_v<NetReceiveCalls_Store>); | ||
static_assert(std::is_trivially_copy_assignable_v<NetReceiveCalls_Store>); | ||
static_assert(std::is_trivially_move_assignable_v<NetReceiveCalls_Store>); | ||
static_assert(std::is_trivially_destructible_v<NetReceiveCalls_Store>); | ||
NetReceiveCalls_Store NetReceiveCalls_global; | ||
|
||
|
||
/** all mechanism instance variables and global variables */ | ||
struct NetReceiveCalls_Instance { | ||
double* c1{}; | ||
double* c2{}; | ||
double* v_unused{}; | ||
double* tsave{}; | ||
const double* node_area{}; | ||
const int* point_process{}; | ||
NetReceiveCalls_Store* global{&NetReceiveCalls_global}; | ||
}; | ||
|
||
|
||
/** connect global (scalar) variables to hoc -- */ | ||
static DoubScal hoc_scalar_double[] = { | ||
{nullptr, nullptr} | ||
}; | ||
|
||
|
||
/** connect global (array) variables to hoc -- */ | ||
static DoubVec hoc_vector_double[] = { | ||
{nullptr, nullptr, 0} | ||
}; | ||
|
||
|
||
static inline int first_pointer_var_index() { | ||
return -1; | ||
} | ||
|
||
|
||
static inline int first_random_var_index() { | ||
return -1; | ||
} | ||
|
||
|
||
static inline int num_net_receive_args() { | ||
return 1; | ||
} | ||
|
||
|
||
static inline int float_variables_size() { | ||
return 4; | ||
} | ||
|
||
|
||
static inline int int_variables_size() { | ||
return 2; | ||
} | ||
|
||
|
||
static inline int get_mech_type() { | ||
return NetReceiveCalls_global.mech_type; | ||
} | ||
|
||
|
||
static inline Memb_list* get_memb_list(NrnThread* nt) { | ||
if (!nt->_ml_list) { | ||
return nullptr; | ||
} | ||
return nt->_ml_list[get_mech_type()]; | ||
} | ||
|
||
|
||
static inline void* mem_alloc(size_t num, size_t size, size_t alignment = 16) { | ||
void* ptr; | ||
posix_memalign(&ptr, alignment, num*size); | ||
memset(ptr, 0, size); | ||
return ptr; | ||
} | ||
|
||
|
||
static inline void mem_free(void* ptr) { | ||
free(ptr); | ||
} | ||
|
||
|
||
static inline void coreneuron_abort() { | ||
abort(); | ||
} | ||
|
||
// Allocate instance structure | ||
static void nrn_private_constructor_NetReceiveCalls(NrnThread* nt, Memb_list* ml, int type) { | ||
assert(!ml->instance); | ||
assert(!ml->global_variables); | ||
assert(ml->global_variables_size == 0); | ||
auto* const inst = new NetReceiveCalls_Instance{}; | ||
assert(inst->global == &NetReceiveCalls_global); | ||
ml->instance = inst; | ||
ml->global_variables = inst->global; | ||
ml->global_variables_size = sizeof(NetReceiveCalls_Store); | ||
} | ||
|
||
// Deallocate the instance structure | ||
static void nrn_private_destructor_NetReceiveCalls(NrnThread* nt, Memb_list* ml, int type) { | ||
auto* const inst = static_cast<NetReceiveCalls_Instance*>(ml->instance); | ||
assert(inst); | ||
assert(inst->global); | ||
assert(inst->global == &NetReceiveCalls_global); | ||
assert(inst->global == ml->global_variables); | ||
assert(ml->global_variables_size == sizeof(NetReceiveCalls_Store)); | ||
delete inst; | ||
ml->instance = nullptr; | ||
ml->global_variables = nullptr; | ||
ml->global_variables_size = 0; | ||
} | ||
|
||
/** initialize mechanism instance variables */ | ||
static inline void setup_instance(NrnThread* nt, Memb_list* ml) { | ||
auto* const inst = static_cast<NetReceiveCalls_Instance*>(ml->instance); | ||
assert(inst); | ||
assert(inst->global); | ||
assert(inst->global == &NetReceiveCalls_global); | ||
assert(inst->global == ml->global_variables); | ||
assert(ml->global_variables_size == sizeof(NetReceiveCalls_Store)); | ||
int pnodecount = ml->_nodecount_padded; | ||
Datum* indexes = ml->pdata; | ||
inst->c1 = ml->data+0*pnodecount; | ||
inst->c2 = ml->data+1*pnodecount; | ||
inst->v_unused = ml->data+2*pnodecount; | ||
inst->tsave = ml->data+3*pnodecount; | ||
inst->node_area = nt->_data; | ||
inst->point_process = ml->pdata; | ||
} | ||
|
||
|
||
|
||
static void nrn_alloc_NetReceiveCalls(double* data, Datum* indexes, int type) { | ||
// do nothing | ||
} | ||
|
||
|
||
void nrn_constructor_NetReceiveCalls(NrnThread* nt, Memb_list* ml, int type) { | ||
#ifndef CORENEURON_BUILD | ||
int nodecount = ml->nodecount; | ||
int pnodecount = ml->_nodecount_padded; | ||
const int* node_index = ml->nodeindices; | ||
double* data = ml->data; | ||
const double* voltage = nt->_actual_v; | ||
Datum* indexes = ml->pdata; | ||
ThreadDatum* thread = ml->_thread; | ||
auto* const inst = static_cast<NetReceiveCalls_Instance*>(ml->instance); | ||
|
||
#endif | ||
} | ||
|
||
|
||
void nrn_destructor_NetReceiveCalls(NrnThread* nt, Memb_list* ml, int type) { | ||
#ifndef CORENEURON_BUILD | ||
int nodecount = ml->nodecount; | ||
int pnodecount = ml->_nodecount_padded; | ||
const int* node_index = ml->nodeindices; | ||
double* data = ml->data; | ||
const double* voltage = nt->_actual_v; | ||
Datum* indexes = ml->pdata; | ||
ThreadDatum* thread = ml->_thread; | ||
auto* const inst = static_cast<NetReceiveCalls_Instance*>(ml->instance); | ||
|
||
#endif | ||
} | ||
|
||
|
||
inline double one_NetReceiveCalls(int id, int pnodecount, NetReceiveCalls_Instance* inst, double* data, const Datum* indexes, ThreadDatum* thread, NrnThread* nt, double v); | ||
inline int increment_c2_NetReceiveCalls(int id, int pnodecount, NetReceiveCalls_Instance* inst, double* data, const Datum* indexes, ThreadDatum* thread, NrnThread* nt, double v); | ||
|
||
|
||
inline int increment_c2_NetReceiveCalls(int id, int pnodecount, NetReceiveCalls_Instance* inst, double* data, const Datum* indexes, ThreadDatum* thread, NrnThread* nt, double v) { | ||
int ret_increment_c2 = 0; | ||
inst->c2[id] = inst->c2[id] + 2.0; | ||
return ret_increment_c2; | ||
} | ||
|
||
|
||
inline double one_NetReceiveCalls(int id, int pnodecount, NetReceiveCalls_Instance* inst, double* data, const Datum* indexes, ThreadDatum* thread, NrnThread* nt, double v) { | ||
double ret_one = 0.0; | ||
ret_one = 1.0; | ||
return ret_one; | ||
} | ||
|
||
|
||
static inline void net_receive_kernel_NetReceiveCalls(double t, Point_process* pnt, NetReceiveCalls_Instance* inst, NrnThread* nt, Memb_list* ml, int weight_index, double flag) { | ||
int tid = pnt->_tid; | ||
int id = pnt->_i_instance; | ||
double v = 0; | ||
int nodecount = ml->nodecount; | ||
int pnodecount = ml->_nodecount_padded; | ||
double* data = ml->data; | ||
double* weights = nt->weights; | ||
Datum* indexes = ml->pdata; | ||
ThreadDatum* thread = ml->_thread; | ||
|
||
inst->tsave[id] = t; | ||
{ | ||
inst->c1[id] = inst->c1[id] + one_NetReceiveCalls(id, pnodecount, inst, data, indexes, thread, nt, v); | ||
increment_c2_NetReceiveCalls(id, pnodecount, inst, data, indexes, thread, nt, v); | ||
} | ||
} | ||
|
||
|
||
static void net_receive_NetReceiveCalls(Point_process* pnt, int weight_index, double flag) { | ||
NrnThread* nt = nrn_threads + pnt->_tid; | ||
Memb_list* ml = get_memb_list(nt); | ||
NetReceiveBuffer_t* nrb = ml->_net_receive_buffer; | ||
if (nrb->_cnt >= nrb->_size) { | ||
realloc_net_receive_buffer(nt, ml); | ||
} | ||
int id = nrb->_cnt; | ||
nrb->_pnt_index[id] = pnt-nt->pntprocs; | ||
nrb->_weight_index[id] = weight_index; | ||
nrb->_nrb_t[id] = nt->_t; | ||
nrb->_nrb_flag[id] = flag; | ||
nrb->_cnt++; | ||
} | ||
|
||
|
||
void net_buf_receive_NetReceiveCalls(NrnThread* nt) { | ||
Memb_list* ml = get_memb_list(nt); | ||
if (!ml) { | ||
return; | ||
} | ||
|
||
NetReceiveBuffer_t* nrb = ml->_net_receive_buffer; | ||
auto* const inst = static_cast<NetReceiveCalls_Instance*>(ml->instance); | ||
int count = nrb->_displ_cnt; | ||
#pragma omp simd | ||
#pragma ivdep | ||
for (int i = 0; i < count; i++) { | ||
int start = nrb->_displ[i]; | ||
int end = nrb->_displ[i+1]; | ||
for (int j = start; j < end; j++) { | ||
int index = nrb->_nrb_index[j]; | ||
int offset = nrb->_pnt_index[index]; | ||
double t = nrb->_nrb_t[index]; | ||
int weight_index = nrb->_weight_index[index]; | ||
double flag = nrb->_nrb_flag[index]; | ||
Point_process* point_process = nt->pntprocs + offset; | ||
net_receive_kernel_NetReceiveCalls(t, point_process, inst, nt, ml, weight_index, flag); | ||
} | ||
} | ||
nrb->_displ_cnt = 0; | ||
nrb->_cnt = 0; | ||
} | ||
|
||
|
||
/** initialize channel */ | ||
void nrn_init_NetReceiveCalls(NrnThread* nt, Memb_list* ml, int type) { | ||
int nodecount = ml->nodecount; | ||
int pnodecount = ml->_nodecount_padded; | ||
const int* node_index = ml->nodeindices; | ||
double* data = ml->data; | ||
const double* voltage = nt->_actual_v; | ||
Datum* indexes = ml->pdata; | ||
ThreadDatum* thread = ml->_thread; | ||
|
||
setup_instance(nt, ml); | ||
auto* const inst = static_cast<NetReceiveCalls_Instance*>(ml->instance); | ||
|
||
if (_nrn_skip_initmodel == 0) { | ||
#pragma omp simd | ||
#pragma ivdep | ||
for (int id = 0; id < nodecount; id++) { | ||
inst->tsave[id] = -1e20; | ||
int node_id = node_index[id]; | ||
double v = voltage[node_id]; | ||
#if NRN_PRCELLSTATE | ||
inst->v_unused[id] = v; | ||
#endif | ||
inst->c1[id] = 0.0; | ||
inst->c2[id] = 0.0; | ||
} | ||
} | ||
} | ||
|
||
|
||
/** register channel with the simulator */ | ||
void _NetReceiveCalls_reg() { | ||
|
||
int mech_type = nrn_get_mechtype("NetReceiveCalls"); | ||
NetReceiveCalls_global.mech_type = mech_type; | ||
if (mech_type == -1) { | ||
return; | ||
} | ||
|
||
_nrn_layout_reg(mech_type, 0); | ||
point_register_mech(mechanism_info, nrn_alloc_NetReceiveCalls, nullptr, nullptr, nullptr, nrn_init_NetReceiveCalls, nrn_private_constructor_NetReceiveCalls, nrn_private_destructor_NetReceiveCalls, first_pointer_var_index(), nullptr, nullptr, 1); | ||
|
||
hoc_register_prop_size(mech_type, float_variables_size(), int_variables_size()); | ||
hoc_register_dparam_semantics(mech_type, 0, "area"); | ||
hoc_register_dparam_semantics(mech_type, 1, "pntproc"); | ||
hoc_register_net_receive_buffering(net_buf_receive_NetReceiveCalls, mech_type); | ||
set_pnt_receive(mech_type, net_receive_NetReceiveCalls, nullptr, num_net_receive_args()); | ||
hoc_register_var(hoc_scalar_double, hoc_vector_double, NULL); | ||
} | ||
} |
Oops, something went wrong.