Skip to content
This repository was archived by the owner on Jul 1, 2025. It is now read-only.

Move quantized embedding ops to OSS. #4606

Closed
wants to merge 1 commit into from
Closed

Move quantized embedding ops to OSS. #4606

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions torch_glow/src/PyTorchModelLoader.cpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -713,8 +713,8 @@ struct EmbeddingBagInputs {
};
};

/// Indexes used for fb::embedding_bag_byte_rowwise_offsets and
/// fb::embedding_bag_4bit_rowwise_offsets inputs.
/// Indexes used for quantized::embedding_bag_byte_rowwise_offsets and
/// quantized::embedding_bag_4bit_rowwise_offsets inputs.
struct EmbeddingBagByteRowwiseOffsetsInputs {
enum {
weight,
Expand Down Expand Up @@ -805,9 +805,9 @@ PyTorchModelLoader::buildSymbolsMapping() {
{{"aten::topk"}, &PyTorchModelLoader::loadTopK},
{{"prim::ConstantChunk"}, &PyTorchModelLoader::loadConstantChunk},
{{"aten::embedding_bag"}, &PyTorchModelLoader::loadEmbeddingBag},
{{"fb::embedding_bag_byte_rowwise_offsets"},
{{"quantized::embedding_bag_byte_rowwise_offsets"},
&PyTorchModelLoader::loadEmbeddingBagByteRowwiseOffsets},
{{"fb::embedding_bag_4bit_rowwise_offsets"},
{{"quantized::embedding_bag_4bit_rowwise_offsets"},
&PyTorchModelLoader::loadEmbeddingBag4BitRowwiseOffsets},
});

Expand Down