JuliaGPU/CUDA.jl

Color the output of `@device_code_sass`

Open

#1,144 opened on Sep 14, 2021

View on GitHub
 (4 comments) (0 reactions) (0 assignees)Julia (274 forks)batch import
enhancementgood first issue

Repository metrics

Stars
 (1,408 stars)
PR merge metrics
 (Avg merge 5d 5h) (16 merged PRs in 30d)

Description

Is your feature request related to a problem? Please describe.

Not a problem per se, but Julia itself does some nice highlighting with @code_native, e.g. image

I find the colouring makes it easier to see what's going on. As it currently is @device_code_sass, has no such colouring.

Describe the solution you'd like

I think it would be rather nice if @device_code_sass similarly colourised its output. Perhaps InteractiveUtils.print_native could be reused? Looking at what @code_native does, I see:

function code_native(io::IO, @nospecialize(f), @nospecialize(types=Tuple);
                     syntax::Symbol=:att, debuginfo::Symbol=:default)
    d = _dump_function(f, types, true, false, false, false, syntax, true, debuginfo)
    if highlighting[:native] && get(io, :color, false)
        print_native(io, d)
    else
        print(io, d)
    end
end

Contributor guide