Skip to content

Instantly share code, notes, and snippets.

View skimo-openhub's full-sized avatar

Sven Verdoolaege skimo-openhub

View GitHub Profile
auto TC = makeMatmulTc();
auto options =
tc::CudaMappingOptions::makeNaiveMappingOptions()
.outerScheduleFusionStrategy(tc::FusionStrategy::Max)
.outerScheduleAllowSkewing(false)
.outerSchedulePositiveOrthant(true)
.intraTileScheduleFusionStrategy(tc::FusionStrategy::Min)
.intraTileScheduleAllowSkewing(false)
.intraTileSchedulePositiveOrthant(true)
.fixParametersBeforeScheduling(false)
09:23:52 Running build/test/test_tc_mapper_bugs
09:23:52 [==========] Running 18 tests from 7 test cases.
09:23:52 [----------] Global test environment set-up.
09:23:52 [----------] 9 tests from TensorDot_32_512_8_2_28_28
09:23:52 [ RUN ] TensorDot_32_512_8_2_28_28.BaseCorrect
09:23:53 [ OK ] Caffe2Test.TcGroupConvolutionOp (4002 ms)
09:23:53 [ RUN ] Caffe2Test.TcConvolutionOp
09:23:56 [ OK ] TensorDot_32_512_8_2_28_28.BaseCorrect (4525 ms)
09:23:56 [ RUN ] TensorDot_32_512_8_2_28_28.ReductionUnroll
09:23:59 [ OK ] Caffe2Test.TcConvolutionOp (6669 ms)