30
30
#include " tc/core/flags.h"
31
31
#include " tc/core/mapping_options.h"
32
32
33
- DEFINE_string (tuner_proto , " " , " Filename to load and store proto cache " );
33
+ DEFINE_string (proto_path , " " , " Filename to load and store proto cache " );
34
34
35
35
TEST (BlockDiagPerm, SimpleAutotune) {
36
36
// 1. Define and setup the TC compilation unit with CUDA memory
@@ -77,7 +77,7 @@ def blockdiagperm2dfissioned_2(float(B, N) I, int32(N) Idx) -> (O) {
77
77
auto options = tc::CudaMappingOptions::makeNaiveCudaMappingOptions ();
78
78
tc::autotune::GeneticAutotunerATen geneticAutotuneATen (tc);
79
79
auto bestOption = geneticAutotuneATen.tune (
80
- FLAGS_tuner_proto , " blockdiagperm2dfissioned_1" , {I, W}, options);
80
+ FLAGS_proto_path , " blockdiagperm2dfissioned_1" , {I, W}, options);
81
81
auto handle = atCompl.compile (
82
82
" blockdiagperm2dfissioned_1" , {I, W}, bestOption.getValue ());
83
83
std::vector<at::Tensor> outputs;
@@ -89,7 +89,7 @@ def blockdiagperm2dfissioned_2(float(B, N) I, int32(N) Idx) -> (O) {
89
89
at::Tensor Idx = at::CPU (at::kInt ).randperm ({500 }).toBackend (at::kCUDA );
90
90
tc::autotune::GeneticAutotunerATen geneticAutotuneATen2 (tc);
91
91
auto bestOption2 = geneticAutotuneATen.tune (
92
- FLAGS_tuner_proto , " blockdiagperm2dfissioned_2" , {O, Idx}, options);
92
+ FLAGS_proto_path , " blockdiagperm2dfissioned_2" , {O, Idx}, options);
93
93
auto handle2 = atCompl.compile (
94
94
" blockdiagperm2dfissioned_2" , {O, Idx}, bestOption2.getValue ());
95
95
std::vector<at::Tensor> outputs2;
@@ -120,7 +120,7 @@ def blockdiagperm2dfissioned_2(float(B, N) I, int32(N) Idx) -> (O) {
120
120
// From root, run with:
121
121
// ./build/examples/blockdiagperm --tuner_threads=10 --tuner_gen_pop_size=10
122
122
// --tuner_gen_generations=3 --tuner_gen_number_elites=4
123
- // --tuner_proto ="/tmp/blockdiagperm"
123
+ // --proto_path ="/tmp/blockdiagperm"
124
124
int main (int argc, char ** argv) {
125
125
::testing::InitGoogleTest (&argc, argv);
126
126
::gflags::ParseCommandLineFlags (&argc, &argv, true );
0 commit comments