aidotse / aidotse/LeakPro

Add HPU support to the time-series MIA example (deferred from #425)

Aberta
#442 1 comentário 0 reações 0 responsáveis Ver no GitHub
Linguagem predominante
Python
Estrelas
23
Forks
28
Merge médio
4d 12h
PRs com merge (30d)
5

Descrição

Follow-up to #425 (HPU support), which was deliberately scoped to exclude this.

The time-series MIA example (`examples/mia/time_series_mia/`) is not in place yet, so HPU enablement for it was left out of #425 rather than being written against a moving target. This issue tracks finishing it once the example itself is settled.

### What #425 already did touch

- `examples/mia/time_series_mia/data_handler.py`, `main.py`, `utils/model_preparation.py` and `main.ipynb` got the mechanical `get_device()` / `mark_step()` swap, so they are not on the old hardcoded-CUDA path.
- `leakpro/signals/utils/get_TS2Vec.py` explicitly falls back to CPU on HPU, because TS2Vec upstream has no HPU support:

```python
# TS2Vec upstream does not support HPU; fall back to CPU for this model.
```

So the example will run on an HPU box, but the representation-learning step silently runs on CPU.

### What still needs doing

1. Decide whether the TS2Vec CPU fallback is acceptable for the DTS attack, or whether the representation step needs an HPU-capable replacement. If it stays on CPU, that belongs in the docs, not just a log line — it's a performance cliff in the middle of an otherwise accelerated run.
2. Validate the example end to end on a Gaudi card once the example is in place, and record which `PT_HPU_LAZY_MODE` it was tested in.
3. Confirm `mark_step()` placement in the time-series training loops matches the pattern used elsewhere (after `optimizer.step()`).

### Not blocking #425

#425 should merge without this. Whoever picks up the time-series example should pick this up with it.

Guia de contribuição

Abrir o guia de contribuição

Avaliação

Esta issue ainda não foi avaliada.

Receba novas issues na sua caixa de entrada

Um resumo curto de issues do GitHub para quem está começando.