mirror of
https://github.com/huggingface/candle.git
synced 2025-06-17 19:18:50 +00:00
Adding embedding op (not generic gather, no select).
This commit is contained in:
@ -346,6 +346,15 @@ impl CudaStorage {
|
||||
}
|
||||
}
|
||||
|
||||
pub(crate) fn embedding_impl(
|
||||
&self,
|
||||
rhs: &Self,
|
||||
hidden_size: usize,
|
||||
vocab_size: usize,
|
||||
) -> Result<Self> {
|
||||
todo!("Implement embedding for gpu");
|
||||
}
|
||||
|
||||
pub(crate) fn matmul_impl(
|
||||
&self,
|
||||
rhs: &Self,
|
||||
|
Reference in New Issue
Block a user