We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 5e3d735 commit c00a882Copy full SHA for c00a882
1 file changed
src/ir.jl
@@ -79,27 +79,23 @@ struct PartialSum <: IR
79
end
80
81
function _get_variables(arg::Mat)
82
- if arg.id isa String
83
- return arg.id
84
- end
85
-
86
- return nothing
+ return _get_variables(arg.id)
87
88
89
function _get_variables(arg::Vec)
90
91
92
93
94
95
96
97
function _get_variables(arg::Scal)
98
99
100
+end
101
102
+function _get_variables(arg::Var)
+ return arg.id
+
+function _get_variables(arg::Const)
+ return arg.value
103
104
105
function _get_variables(arg::Real)
0 commit comments