bug fixes for silu_back
This commit is contained in:
parent
9d6fc28f18
commit
6fb08b4554
1 changed files with 2 additions and 2 deletions
4
ggml.c
4
ggml.c
|
@ -5601,7 +5601,7 @@ struct ggml_tensor * ggml_silu_back(
|
||||||
bool is_node = false;
|
bool is_node = false;
|
||||||
|
|
||||||
if (a->grad || b->grad) {
|
if (a->grad || b->grad) {
|
||||||
GGML_ASSERT(false); // TODO: implement backward
|
// TODO: implement backward
|
||||||
is_node = true;
|
is_node = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -8905,7 +8905,7 @@ static void ggml_compute_forward_silu_back(
|
||||||
switch (src0->type) {
|
switch (src0->type) {
|
||||||
case GGML_TYPE_F32:
|
case GGML_TYPE_F32:
|
||||||
{
|
{
|
||||||
ggml_compute_forward_silu_back_f32(params, grad, src0, dst);
|
ggml_compute_forward_silu_back_f32(params, src0, grad, dst);
|
||||||
} break;
|
} break;
|
||||||
default:
|
default:
|
||||||
{
|
{
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue