Skip to content

Commit 026020e

Browse files
committed
Update on "decompose con1d to conv2d for cuda backend"
this diff decomposed conv1d into conv2d for aoti-cuda backend support Differential Revision: [D84296877](https://our.internmc.facebook.com/intern/diff/D84296877/) [ghstack-poisoned]
2 parents ae17932 + 5351655 commit 026020e

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

examples/models/toy_model/__init__.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,19 @@
44
# This source code is licensed under the BSD-style license found in the
55
# LICENSE file in the root directory of this source tree.
66

7-
from .model import AddModule, AddMulModule, LinearModule, MulModule, SoftmaxModule
7+
from .model import (
8+
AddModule,
9+
AddMulModule,
10+
Conv1dModule,
11+
LinearModule,
12+
MulModule,
13+
SoftmaxModule,
14+
)
815

916
__all__ = [
1017
AddModule,
1118
AddMulModule,
19+
Conv1dModule,
1220
LinearModule,
1321
MulModule,
1422
SoftmaxModule,

0 commit comments

Comments
 (0)