Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Menu
Open sidebar
monticore
EmbeddedMontiArc
generators
CNNArch2Gluon
Commits
ab7434d0
Commit
ab7434d0
authored
Aug 12, 2019
by
Sebastian Nickels
Browse files
Moved some MXNet-related code from CNNArch2X to here
parent
c62cdf1e
Pipeline
#170413
failed with stages
in 2 minutes and 18 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
src/main/java/de/monticore/lang/monticar/cnnarch/gluongenerator/CNNArch2Gluon.java
View file @
ab7434d0
...
...
@@ -20,15 +20,14 @@
*/
package
de.monticore.lang.monticar.cnnarch.gluongenerator
;
import
de.monticore.lang.monticar.cnnarch._symboltable.IODeclarationSymbol
;
import
de.monticore.lang.monticar.cnnarch._symboltable.VariableDeclarationSymbol
;
import
de.monticore.lang.monticar.cnnarch._symboltable.VariableSymbol
;
import
de.monticore.lang.monticar.cnnarch.generator.CNNArchGenerator
;
import
de.monticore.lang.monticar.cnnarch.generator.Target
;
import
de.monticore.lang.monticar.cnnarch.generator.TemplateConfiguration
;
import
de.monticore.lang.monticar.cnnarch._symboltable.ArchitectureSymbol
;
import
de.se_rwth.commons.logging.Log
;
import
de.monticore.lang.monticar.generator.FileContent
;
import
de.monticore.lang.monticar.generator.cmake.CMakeConfig
;
import
de.monticore.lang.monticar.generator.cmake.CMakeFindModule
;
import
java.util.*
;
...
...
@@ -112,4 +111,20 @@ public class CNNArch2Gluon extends CNNArchGenerator {
architecture
,
templateConfiguration
);
return
compilePythonFiles
(
archTc
,
architecture
);
}
public
Map
<
String
,
String
>
generateCMakeContent
(
String
rootModelName
)
{
// model name should start with a lower case letter. If it is a component, replace dot . by _
rootModelName
=
rootModelName
.
replace
(
'.'
,
'_'
).
replace
(
'['
,
'_'
).
replace
(
']'
,
'_'
);
rootModelName
=
rootModelName
.
substring
(
0
,
1
).
toLowerCase
()
+
rootModelName
.
substring
(
1
);
CMakeConfig
cMakeConfig
=
new
CMakeConfig
(
rootModelName
);
cMakeConfig
.
addModuleDependency
(
new
CMakeFindModule
(
"Armadillo"
,
true
));
cMakeConfig
.
addCMakeCommand
(
"set(LIBS ${LIBS} mxnet)"
);
Map
<
String
,
String
>
fileContentMap
=
new
HashMap
<>();
for
(
FileContent
fileContent
:
cMakeConfig
.
generateCMakeFiles
()){
fileContentMap
.
put
(
fileContent
.
getFileName
(),
fileContent
.
getFileContent
());
}
return
fileContentMap
;
}
}
\ No newline at end of file
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment