196d6f5dc1
Add `device` parameter to `Linear` subclasses and `Embedding` |
||
---|---|---|
.. | ||
__init__.py | ||
modules.py | ||
triton_based_modules.py |
196d6f5dc1
Add `device` parameter to `Linear` subclasses and `Embedding` |
||
---|---|---|
.. | ||
__init__.py | ||
modules.py | ||
triton_based_modules.py |