ayo/ipex-llm
1
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions 1

Nano: add patching function for dtype (#6947)

Browse source 
* add patching function for dtype

* update

* fix pep8

* update the parameter name
This commit is contained in:
Junwei Deng 2022-12-16 16:16:12 +08:00 committed by GitHub
parent 6b65968e3c
commit f304d33168
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
docs/readthedocs/source/doc/PythonAPI/Nano/pytorch.rst
View file

@ -33,3 +33,11 @@ Patch API
:members:
:undoc-members:
:show-inheritance:
.. automodule:: bigdl.nano.pytorch.patching.gpu_cpu.gpu_cpu
:members: patch_cuda, unpatch_cuda
:show-inheritance:
.. automodule:: bigdl.nano.pytorch.patching.dtype_patching.dtype_patching
:members: patch_dtype
:show-inheritance: