Skip to content
This repository has been archived by the owner on Nov 25, 2024. It is now read-only.

[BUG] WholeMemoryTensor Gather References Nonexistent Property #168

Open
alexbarghi-nv opened this issue May 20, 2024 · 0 comments
Open

Comments

@alexbarghi-nv
Copy link
Member

I believe this should be self.dtype but it is self.embedding_tensor.dtype. This causes an exception to be raised whenever the dtype is not provided.

force_dtype if force_dtype is not None else self.embedding_tensor.dtype

rapids-bot bot pushed a commit that referenced this issue May 28, 2024
A quick fixes to this issue (#168). Set correct default wholememory tensor gather results data type.

Authors:
  - https://github.com/linhu-nv

Approvers:
  - Chuang Zhu (https://github.com/chuangz0)

URL: #173
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant