Skip to content

[MTHREADS] adapt some config to mt backend#2210

Open
Oslomayor wants to merge 1 commit intoflagos-ai:masterfrom
Oslomayor:fix/mthreads-adapt-202604021534
Open

[MTHREADS] adapt some config to mt backend#2210
Oslomayor wants to merge 1 commit intoflagos-ai:masterfrom
Oslomayor:fix/mthreads-adapt-202604021534

Conversation

@Oslomayor
Copy link
Copy Markdown
Contributor

@Oslomayor Oslomayor commented Apr 2, 2026

Adapted from upstream commits:
- 1 tune config: conj_physical (new operator tune config)
- 3 custom operator bug fixes: arange (int64 float params), repeat (zero-dim check), repeat_interleave (empty repeats handling)

Files modified: 4

Adapt details:
- conj_physical: added tune config to _mthreads/tune_configs.yaml
- arange: fixed int64 dtype with float parameters (int conversion for start/end/step)
- repeat: fixed zero-dim bug (in0_shape[i] * sizes_shape[i] == 0)
- repeat_interleave: added empty repeats handling (repeats.numel() == 0)

Above description is written by agent.

Adapted from upstream commits:
- 1 tune config: conj_physical (new operator tune config)
- 3 custom operator bug fixes: arange (int64 float params), repeat (zero-dim check), repeat_interleave (empty repeats handling)

Files modified: 4
Items skipped: 64 (no mthreads custom impl or not needed)

Adapt details:
- conj_physical: added tune config to _mthreads/tune_configs.yaml
- arange: fixed int64 dtype with float parameters (int conversion for start/end/step)
- repeat: fixed zero-dim bug (in0_shape[i] * sizes_shape[i] == 0)
- repeat_interleave: added empty repeats handling (repeats.numel() == 0)

Skipped items:
- index_put: mthreads has independent impl, main ops refactor not affected
- one_hot: mthreads has optimized impl (specialized kernels), main ops refactor not affected
- other 64 ops: no mthreads custom impl or only format/debug changes
@Oslomayor Oslomayor force-pushed the fix/mthreads-adapt-202604021534 branch from 54d8f36 to dd17ead Compare April 2, 2026 09:00
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant