Skip to content

Commit 0f5815d

Browse files
committed
Fixes from review.
1 parent 12aa00d commit 0f5815d

2 files changed

Lines changed: 8 additions & 2 deletions

File tree

src/cuTile.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ include("language/math.jl")
3838
include("language/operations.jl")
3939
include("language/atomics.jl")
4040

41-
public launch, Tiled, ByTarget, @compiler_options, var"@."
41+
public launch, Tiled, ByTarget, @compiler_options, @.
4242
launch(args...) = error("Please import CUDA.jl before using `cuTile.launch`.")
4343

4444
"""
@@ -88,7 +88,7 @@ the broadcast through cuTile kernels.
8888
ct.@. C = A + sin(B)
8989
# equivalent to: Tiled(C) .= Tiled(A) .+ sin.(Tiled(B))
9090
"""
91-
macro var"."(ex)
91+
macro __dot__(ex)
9292
esc(_wrap_tiled(Base.Broadcast.__dot__(ex)))
9393
end
9494

test/execution/broadcast.jl

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -885,6 +885,12 @@ end
885885
@test Array(C) Array(A) .+ Array(B)
886886
end
887887

888+
@testset "ct.@. expands to Tiled" begin
889+
ex = @macroexpand ct.@. C = A + B
890+
# The macro should produce Tiled() wrapping, not plain dotted calls
891+
@test occursin("Tiled", string(ex))
892+
end
893+
888894
@testset "ct.@. in-place" begin
889895
n = 1024
890896
A = CUDA.rand(Float32, n)

0 commit comments

Comments
 (0)