Revert embedding dataformat cast (#59)

We don't need explicit embedding dataformat cast in tvm (from float32 to
bf16) as dataformat workaround for this case is implemented in mlir.

PRs for reference:

- [TVM cast workaround](#55)
- [Embedding Op
workaround](tenstorrent/tt-mlir#1583)
- [EmbeddingBackward Op
workaround](tenstorrent/tt-mlir#1756)

Related to this issue
tenstorrent/tt-forge-fe#1112

python/tvm/relay/frontend/pytorch.py

@@ -2700,10 +2700,7 @@ def embedding(self, inputs, input_types):
             #  exposes a few bugs in tt-mlir https://github.com/tenstorrent/tt-mlir/issues/1215
             logger.warning("Casting input indices of embedding op from {} to int32", indicies_dtype)
             indices = tvm.relay.cast(indices, "int32")
-        # cast the weight to bfloat16 if it is float32
-        if weight.type_annotation.dtype == "float32":
-            weight = tvm.relay.cast(weight, "bfloat16")
-        return tvm.relay.cast(_op.embedding(weight, indices, axis=0), "float32")
+        return _op.embedding(weight, indices, axis=0)
 
     def embedding_bag(self, inputs, input_types):