Skip to content

Commit 684b4ce

Browse files
committed
issue/889 - revert embedding modifications
1 parent 6f8a443 commit 684b4ce

7 files changed

Lines changed: 0 additions & 219 deletions

File tree

include/infinicore/ops.hpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
#include "ops/add_rms_norm.hpp"
55
#include "ops/attention.hpp"
66
#include "ops/causal_softmax.hpp"
7-
#include "ops/embedding.hpp"
87
#include "ops/flash_attention.hpp"
98
#include "ops/kv_caching.hpp"
109
#include "ops/matmul.hpp"

include/infinicore/ops/embedding.hpp

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,6 @@
44

55
namespace infinicore::op {
66

7-
class Embedding {
8-
public:
9-
using schema = void (*)(Tensor, Tensor, Tensor);
10-
static void execute(Tensor out, Tensor input, Tensor weight);
11-
static common::OpDispatcher<schema> &dispatcher();
12-
};
13-
147
Tensor embedding(Tensor input, Tensor weight);
158
void embedding_(Tensor out, Tensor input, Tensor weight);
169
} // namespace infinicore::op

include/infiniop.h

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
#include "infiniop/ops/clip.h"
1010
#include "infiniop/ops/conv.h"
1111
#include "infiniop/ops/dequantize_awq.h"
12-
#include "infiniop/ops/embedding.h"
1312
#include "infiniop/ops/flash_attention.h"
1413
#include "infiniop/ops/gelu.h"
1514
#include "infiniop/ops/gemm.h"

include/infiniop/ops/embedding.h

Lines changed: 0 additions & 25 deletions
This file was deleted.

src/infinicore/ops/embedding/embedding.cc

Lines changed: 0 additions & 47 deletions
This file was deleted.

src/infinicore/ops/embedding/embedding_infiniop.cc

Lines changed: 0 additions & 49 deletions
This file was deleted.

src/infiniop/ops/embedding/operator.cc

Lines changed: 0 additions & 89 deletions
This file was deleted.

0 commit comments

Comments
 (0)