intel / intel/torch-xpu-ops

Implement torch.linalg.lu_factor without pivoting

Open
#3,951 0 comments 0 reactions 1 assignee Assigned to @jkosnox View on GitHub
module: ops
Dominant language
Python
Stars
113
Forks
128
Avg merge
5d 13h
Merged PRs (30d)
107

Description

### 🚀 The feature, motivation and pitch

Currently `torch.linalg.lu_factor` is enabled via oneMKL but if `pivot==False` it throws error. Op can't fallback to CPU, as CPU doesn't have such implementation either.

Goal of this task is to add support for lu_factor without pivoting. oneMKL `getrfnp`/`getrfnp_batch` should be used

### Alternatives

_No response_

### Additional context

Gap was found in unit tests, not model

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.