shijieliu
shijieliu
Description: Hi Joey, One of the customers, Varian, as mentioned in the email are using Pytorch and they asked if Pytorch export capability is available on HugeCTR. I can get...
This PR is to enable SOK lookup support DeepRec EmbeddingVariableGPU. It includes 4 parts: 1. move EmbeddingVarGPU from kernel to framework 2. add hugectr as a submodule under SOK 3....
# Current behavior ``` 2022-10-19 12:39:39.948019: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0 2022-10-19 12:39:39.948020: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0 INFO:tensorflow:Parsing ../data//train.csv INFO:tensorflow:Parsing ../data//train.csv WARNING:tensorflow:The default value of...
Hi, team, In the `ShardedEmbeddingBagCollection`, I found torchrec explicit make dp lookup as `DistributedDataParallel`([code here](https://github.com/pytorch/torchrec/blob/main/torchrec/distributed/embeddingbag.py#L503)). And I also know inside [DistributedModelParallel](https://github.com/pytorch/torchrec/blob/main/torchrec/distributed/model_parallel.py#L216) we have ddp wrapper to warp the non-sharded part...