Skip to content

fix/rocm properties export#487

Open
GitGlimpse895 wants to merge 4 commits intohuggingface:mainfrom
GitGlimpse895:fix/rocm-properties-export
Open

fix/rocm properties export#487
GitGlimpse895 wants to merge 4 commits intohuggingface:mainfrom
GitGlimpse895:fix/rocm-properties-export

Conversation

@GitGlimpse895
Copy link
Copy Markdown

ROCMProperties was fully implemented in device.py with its own docstring and example, but was never wired up end-to-end:

  • Device.properties was typed as CUDAProperties | None, silently rejecting ROCMProperties at the type-system level.
  • Device.__post_init__ validated CUDAProperties placement but had no symmetric guard for ROCMProperties, allowing invalid combinations like Device(type="cuda", properties=ROCMProperties(...)) to pass silently.
  • ROCMProperties was never exported from kernels/layer/__init__.py or the top-level kernels/__init__.py, making from kernels import ROCMProperties fail despite being shown in the class docstring example.

This PR fixes all three gaps across three files, completing the ROCm capability-range kernel selection feature that was already wired up in _ROCMRepos.

@GitGlimpse895 GitGlimpse895 force-pushed the fix/rocm-properties-export branch 4 times, most recently from 65ef442 to ae0a6eb Compare April 23, 2026 01:42
@HuggingFaceDocBuilderDev
Copy link
Copy Markdown

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

@GitGlimpse895 GitGlimpse895 force-pushed the fix/rocm-properties-export branch from 441d889 to f98c3ec Compare April 23, 2026 13:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants