coffee/DeepInfer
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepInfer.git synced 2026-10-29 07:17:08 +08:00
Code Issues Packages Projects Releases Wiki Activity
DeepInfer/deepinfer/neuron_ops
History
whn09 d2389f8a9d Neuron deep_ep: on-device fixes from the first NeuronCore run 
- Engram: gather rows where the table lives (host by default, read via UVA
  on GPU) and move them to the compute device.
- EPHandle: the topk_idx version guard tolerates inference-mode tensors on
  the neuron device that do not report is_inference().

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>
2026-10-20 21:35:48 +08:00
..
vendor Neuron deep_ep: on-device fixes from the first NeuronCore run 2026-10-20 21:35:48 +08:00
__init__.py Add AWS Trainium (Neuron) platform, basic mode — part 1 2026-10-20 18:34:51 +08:00
NEURON_PORT.md Add AWS Trainium (Neuron) platform, basic mode — part 1 2026-10-20 18:34:51 +08:00
transfer_to_neuron.py Neuron basic mode — part 2: deep_ep, sampling kernels, fixes 2026-10-20 18:57:27 +08:00