File tree Expand file tree Collapse file tree 1 file changed +12
-5
lines changed
test/Microsoft.ML.OnnxTransformerTest Expand file tree Collapse file tree 1 file changed +12
-5
lines changed Original file line number Diff line number Diff line change @@ -54,10 +54,14 @@ public DnnImageFeaturizerTests(ITestOutputHelper helper) : base(helper)
54
54
}
55
55
56
56
[ OnnxFact ]
57
- //Skipping test temporarily. This test will be re-enabled once the cause of failures has been determined
58
- [ Trait ( "Category" , "SkipInCI" ) ]
59
57
public void TestDnnImageFeaturizer ( )
60
58
{
59
+ //skip running for x86 as this test using too much memory (over 2GB limit on x86)
60
+ //and very like to hit memory related issue when running on CI
61
+ //TODO: optimized memory usage in related code and enable x86 test run
62
+ if ( ! Environment . Is64BitProcess )
63
+ return ;
64
+
61
65
var samplevector = GetSampleArrayData ( ) ;
62
66
63
67
var dataView = DataViewConstructionUtils . CreateFromList ( Env ,
@@ -125,12 +129,15 @@ public void OnnxFeaturizerWorkout()
125
129
}
126
130
}
127
131
128
- // Onnx is only supported on x64 Windows
129
132
[ OnnxFact ]
130
- //Skipping test temporarily. This test will be re-enabled once the cause of failures has been determined
131
- [ Trait ( "Category" , "SkipInCI" ) ]
132
133
public void TestOldSavingAndLoading ( )
133
134
{
135
+ //skip running for x86 as this test using too much memory (over 2GB limit on x86)
136
+ //and very like to hit memory related issue when running on CI
137
+ //TODO: optimized memory usage in related code and enable x86 run
138
+ if ( ! Environment . Is64BitProcess )
139
+ return ;
140
+
134
141
var samplevector = GetSampleArrayData ( ) ;
135
142
136
143
var dataView = ML . Data . LoadFromEnumerable (
You can’t perform that action at this time.
0 commit comments