lhnguyen102 / cuTAGI

CUDA implementation of Tractable Approximate Gaussian Inference

MIT License

24 stars 9 forks source link

Multi-embedding Layer #35

Closed lhnguyen102 closed 10 months ago

lhnguyen102 commented 10 months ago

Description

This PR introduces the embedding layer for CPU version

Changes Made

Added embedding_cpu.cpp
Added embedding_cpu.h
Added embedding folder including test script for embedding layer
Added layer code and prop to struct_var.h
Added reset_zero functionality for DeltaParam class in data_transfer_cpu.cpp to set all delta values to zero

Reviewer Note

Run the test using the following command

build/main test_embedding

jamesgoulet commented 10 months ago

👌