File tree Expand file tree Collapse file tree 1 file changed +17
-14
lines changed Expand file tree Collapse file tree 1 file changed +17
-14
lines changed Original file line number Diff line number Diff line change @@ -272,29 +272,32 @@ def test_debugprint():
272
272
print_view_map = True ,
273
273
)
274
274
s = s .getvalue ()
275
+ Gemv_op_name = "CGemv" if pytensor .config .blas__ldflags else "Gemv"
275
276
exp_res = dedent (
276
277
r"""
277
278
Composite{(i2 + (i0 - i1))} 4
278
- ├─ ExpandDims{axis=0} v={0: [0]} 3
279
- │ └─ CGemv{inplace} d={0: [0]} 2
280
- │ ├─ AllocEmpty{dtype='float64'} 1
281
- │ │ └─ Shape_i{0} 0
282
- │ │ └─ B
283
- │ ├─ 1.0
284
- │ ├─ B
285
- │ ├─ <Vector(float64, shape=(?,))>
286
- │ └─ 0.0
287
- ├─ D
288
- └─ A
279
+ ├─ ExpandDims{axis=0} v={0: [0]} 3
280
+ """
281
+ f" │ └─ { Gemv_op_name } {{inplace}} d={{0: [0]}} 2"
282
+ r"""
283
+ │ ├─ AllocEmpty{dtype='float64'} 1
284
+ │ │ └─ Shape_i{0} 0
285
+ │ │ └─ B
286
+ │ ├─ 1.0
287
+ │ ├─ B
288
+ │ ├─ <Vector(float64, shape=(?,))>
289
+ │ └─ 0.0
290
+ ├─ D
291
+ └─ A
289
292
290
293
Inner graphs:
291
294
292
295
Composite{(i2 + (i0 - i1))}
293
- ← add 'o0'
296
+ ← add 'o0'
294
297
├─ i2
295
298
└─ sub
296
- ├─ i0
297
- └─ i1
299
+ ├─ i0
300
+ └─ i1
298
301
"""
299
302
).lstrip ()
300
303
You can’t perform that action at this time.
0 commit comments