Skip to content

Conversation

@ColinPeppler
Copy link
Contributor

@ColinPeppler ColinPeppler commented Dec 20, 2025

Added a disclaimer that ptx, cubin, and mlir attributes are only populated when their respective CUTE_DSL_KEEP_* envvar are set.

print(f"PTX: {compiled_foo.__ptx__}")    # Otherwise, this prints None without CUTE_DSL_KEEP_PTX=1

@ColinPeppler
Copy link
Contributor Author

Hi @Junkai-Wu @hwu36, I was initially confused why compiled.__ptx__ wasn't showing anything. I'm wondering if we're required to set CUTE_DSL_KEEP_PTX=1 for __ptx__ to be populated.

Copy link
Collaborator

@fengxie fengxie left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

@brandon-yujie-sun can you double check?

- ``__cubin__``: The generated CUBIN data of the compiled kernel.
- ``__mlir__``: The generated IR code of the compiled kernel.

These attributes are populated only when the corresponding ``CUTE_DSL_KEEP_*`` environment variable is enabled;
Copy link
Collaborator

@brandon-yujie-sun brandon-yujie-sun Jan 9, 2026

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Yes, these attributes are meaningful only when the corresponding dumping is enabled, either by setting ENV VAR or through compile options for cute.compile, see https://github.com/NVIDIA/cutlass/blob/main/media/docs/pythonDSL/cute_dsl_general/dsl_jit_compilation_options.rst

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.

3 participants