@operation -> @device

Author: Roger-luo

src/YaoCompiler.jl

@@ -1,6 +1,6 @@
 module YaoCompiler
 
-export @operation,
+export @device,
     @gate,
     @ctrl,
     @measure,

src/compiler/intrinsics.jl

@@ -16,7 +16,7 @@ export main, apply, measure, barrier, expect
 @intrinsic_stub device expect(r::AbstractRegister, ::Locations, nshots::Int)
 
 # this is just for doing overlay since we don't
-# have explicit register semantic inside @operation
+# have explicit register semantic inside @device
 @intrinsic_stub device apply(gate::Routine)
 @intrinsic_stub device apply(gate::Routine, ::Locations)
 @intrinsic_stub device apply(gate::Routine, ::Locations, ::CtrlLocations)

src/compiler/syntax.jl

@@ -4,11 +4,11 @@ xctrl_locations(ex) = Expr(:call, :($YaoLocations.CtrlLocations), ex)
 """
     @gate <locs> => <gate>
 
-Syntax sugar for `apply(gate, locs)`, must be used inside `@operation`.
-See also [`@operation`](@ref).
+Syntax sugar for `apply(gate, locs)`, must be used inside `@device`.
+See also [`@device`](@ref).
 
 !!! tips
-    You don't have to write `@gate` in most cases inside `@operation`.
+    You don't have to write `@gate` in most cases inside `@device`.
     But in case there is ambuigity, you can annotate the expression
     with `@gate` explicitly.
 """
@@ -22,8 +22,8 @@ end
 """
     @ctrl <ctrl_locs> <locs> => <gate>
 
-Syntax sugar for `apply(gate, locs, ctrl_locs)`, must be used inside `@operation`.
-See also [`@operation`](@ref).
+Syntax sugar for `apply(gate, locs, ctrl_locs)`, must be used inside `@device`.
+See also [`@device`](@ref).
 """
 macro ctrl(ctrl_locs, ex::Expr)
     @match ex begin
@@ -53,12 +53,12 @@ macro barrier(locs)
 end
 
 """
-    @operation <function def>
+    @device <function def>
 
-Annotate a Julia function as YaoLang operation kernel.
+Annotate a Julia function as YaoLang device kernel.
 """
-macro operation(ex)
-    esc(operation_m(__module__, ex))
+macro device(ex)
+    esc(device_m(__module__, ex))
 end
 
 """
@@ -71,7 +71,7 @@ on them directly.
 """
 function routine_stub end
 
-function operation_m(mod::Module, ex)
+function device_m(mod::Module, ex)
     is_function(ex) || error("expect a function definition")
     jlfn = JLFunction(ex)
     isnothing(jlfn.kwargs) || error("kwargs is not supported")
@@ -86,7 +86,7 @@ function codegen_routine(jlfn::JLFunction)
 
     return quote
         $(codegen_routine_type(jlfn, typename))
-        $(codegen_operation(jlfn, typename))
+        $(codegen_device(jlfn, typename))
         $(codegen_inference_limit_heuristics(jlfn, typename))
         $(codegen_routine_stub(jlfn, typename))
         $(codegen_binding(jlfn, typename))
@@ -112,7 +112,7 @@ function codegen_binding(def::JLFunction, typename)
     end
 end
 
-function codegen_operation(def::JLFunction, typename)
+function codegen_device(def::JLFunction, typename)
     self = gensym(:self)
     args = name_only.(def.args)
 
@@ -207,17 +207,17 @@ function transpile_gate_syntax(ex)
         :($locs => $gate) => xcall(GlobalRef(Intrinsics, :apply), gate, xlocations(locs))
         # this will appear in anonymous function definition
         # TODO: disambuigity this and function contains only single line
-        # @operation function circuit(theta, phi)
+        # @device function circuit(theta, phi)
         #     1 => X
         # end
         # Expr(:block, stmt1, line::LineNumberNode, stmt2) => ex
-        Expr(:macrocall, Symbol("@operation"), _...) => error("syntax: cannot have nested @operation")
+        Expr(:macrocall, Symbol("@device"), _...) => error("syntax: cannot have nested @device")
         Expr(:macrocall, name, args...) => begin
             if is_syntax_macro(name)
                 return ex
             else
                 # we force top scope locs=>gate to be treated as gate stmt inside
-                # all @operation region including non-YaoCompiler macros to make things
+                # all @device region including non-YaoCompiler macros to make things
                 # like @inbounds etc. work
                 return Expr(:macrocall, name, map(transpile_gate_syntax, args)...)
             end
@@ -239,8 +239,8 @@ end
     ci, nargs = obtain_codeinfo(op)
     new = NewCodeInfo(ci)
     register = insert!(new.slots, 2, Symbol("#register#"))
-    operation = insert!(new.slots, 3, Symbol("#op#"))
-    unpack_operation!(new, operation, nargs)
+    device = insert!(new.slots, 3, Symbol("#op#"))
+    unpack_device!(new, device, nargs)
     for (v, stmt) in new
         @switch stmt begin
             @case Expr(:call, GlobalRef(&Intrinsics, name), args...)
@@ -278,9 +278,9 @@ end
     ci, nargs = obtain_codeinfo(op)
     new = NewCodeInfo(ci)
     register = insert!(new.slots, 2, Symbol("#register#"))
-    operation = insert!(new.slots, 3, Symbol("#op#"))
+    device = insert!(new.slots, 3, Symbol("#op#"))
     glob_locs = insert!(new.slots, 4, Symbol("#locs#"))
-    unpack_operation!(new, operation, nargs)
+    unpack_device!(new, device, nargs)
 
     for (v, stmt) in new
         _update_slot_stmt(new, v, stmt) do stmt
@@ -327,10 +327,10 @@ end
     ci, nargs = obtain_codeinfo(op)
     new = NewCodeInfo(ci)
     register = insert!(new.slots, 2, Symbol("#register#"))
-    operation = insert!(new.slots, 3, Symbol("#op#"))
+    device = insert!(new.slots, 3, Symbol("#op#"))
     glob_locs = insert!(new.slots, 4, Symbol("#locs#"))
     glob_ctrl = insert!(new.slots, 5, Symbol("#ctrl#"))
-    unpack_operation!(new, operation, nargs)
+    unpack_device!(new, device, nargs)
 
     for (v, stmt) in new
         _update_slot_stmt(new, v, stmt) do stmt
@@ -398,7 +398,7 @@ function obtain_codeinfo(::Type{Operation{P,Args}}) where {P,Args}
     return ci, nargs
 end
 
-function unpack_operation!(new::NewCodeInfo, op::NewSlotNumber, nargs::Int)
+function unpack_device!(new::NewCodeInfo, op::NewSlotNumber, nargs::Int)
     # %parent = op.parent
     parent = push!(new, Expr(:call, GlobalRef(Base, :getfield), op, QuoteNode(:parent)))
 

src/compiler/types.jl

@@ -78,7 +78,7 @@ end
 end
 
 @noinline function measure_cmp(lhs, rhs)
-    throw(IntrinsicError("cannot compare measurement result outside @operation"))
+    throw(IntrinsicError("cannot compare measurement result outside @device"))
 end
 
 # struct Chain

test/emulate.jl

@@ -58,7 +58,7 @@ end
     return MeasureResult{typeof(result)}(result)
 end
 
-@operation function test_intrinsic(theta, phi)
+@device function test_intrinsic(theta, phi)
     # syntax sugar
     1 => X
     @gate 2 => Z
@@ -67,7 +67,7 @@ end
     return
 end
 
-@operation function test_location_map(theta)
+@device function test_location_map(theta)
     3:6 => test_intrinsic(theta, 2.5)
     @ctrl 1 3:6 => test_intrinsic(theta, 2.1)
     return

test/interp.jl

@@ -8,7 +8,7 @@ using YaoCompiler.Intrinsics
 using CompilerPluginTools
 using YaoCompiler.Intrinsics: measure
 
-@operation function test_basic(theta, phi)
+@device function test_basic(theta, phi)
     # syntax sugar
     1 => X
     @gate 2 => Z
@@ -22,15 +22,15 @@ using YaoCompiler.Intrinsics: measure
     return (a = a, b = c)
 end
 
-@operation function test_pure_quantum()
+@device function test_pure_quantum()
     ret = @gate 1:4 => test_basic(1.0, 2.0)
     @ctrl 2 1 => Rx(2.2)
     return ret
 end
 
 ci, type = @yao_code_typed(test_pure_quantum())[1]
 
-@operation function routine2(theta)
+@device function routine2(theta)
     2 => Rx(theta)
 end
 

test/printing.jl

@@ -4,7 +4,7 @@ using YaoLocations
 using YaoArrayRegister
 using YaoCompiler.Intrinsics
 
-@operation function test_intrinsic(theta, phi)
+@device function test_intrinsic(theta, phi)
     # syntax sugar
     1 => X
     @gate 2 => Z

test/zx.jl

@@ -10,7 +10,7 @@ using YaoHIR
 using ZXCalculus
 using YaoArrayRegister
 
-@operation function test_a()
+@device function test_a()
     1 => X
     1 => X
 end
@@ -26,7 +26,7 @@ ir, = code_typed(Intrinsics.apply, (AnyReg, typeof(test_a())); interp)[1]
     Expr(:invoke, _, apply, _, &(QuoteNode(H)), &(QuoteNode(Locations(1))))::Nothing
 end
 
-@operation function test_b()
+@device function test_b()
     1 => Rx(2.0)
     1 => Rz(3.0)
 end
@@ -40,7 +40,7 @@ ir, = code_typed(Intrinsics.apply, (AnyReg, typeof(test_b())); interp)[1]
     Expr(:invoke, _, apply, _, &(QuoteNode(Rz(3.0))), &(QuoteNode(Locations(1))))::Nothing
 end
 
-@operation function test_cir()
+@device function test_cir()
     5 => H
     @ctrl 4 5 => X
     5 => shift(7 / 4 * π)