xref: /aosp_15_r20/external/pytorch/scripts/proto.ps1 (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker param(
2*da0073e9SAndroid Build Coastguard Worker   [string]$protoc,
3*da0073e9SAndroid Build Coastguard Worker   [string]$srcdir,
4*da0073e9SAndroid Build Coastguard Worker   [string]$unprocessed,
5*da0073e9SAndroid Build Coastguard Worker   [string]$processed,
6*da0073e9SAndroid Build Coastguard Worker   [string]$out
7*da0073e9SAndroid Build Coastguard Worker )
8*da0073e9SAndroid Build Coastguard Worker $ErrorActionPreference = "Stop"
9*da0073e9SAndroid Build Coastguard Worker Get-Content $unprocessed | % {$_ -Replace "caffe2/proto/caffe2.proto", "caffe2.proto"} | Set-Content $processed
10*da0073e9SAndroid Build Coastguard Worker Add-Content -Path $processed -Value "option optimize_for = LITE_RUNTIME;`n" -NoNewline
11*da0073e9SAndroid Build Coastguard Worker $dir = (Get-Item $processed).DirectoryName
12*da0073e9SAndroid Build Coastguard Worker 
13*da0073e9SAndroid Build Coastguard Worker copy $srcdir/caffe2/proto/caffe2.proto $srcdir/caffe2.proto
14*da0073e9SAndroid Build Coastguard Worker Add-Content -Path $srcdir/caffe2.proto -Value "option optimize_for = LITE_RUNTIME;`n" -NoNewline
15*da0073e9SAndroid Build Coastguard Worker 
16*da0073e9SAndroid Build Coastguard Worker $processed = (Get-Item $processed).Name
17*da0073e9SAndroid Build Coastguard Worker $cmd = "$protoc -I${dir} --cpp_out=$out $processed"
18*da0073e9SAndroid Build Coastguard Worker Invoke-Expression $cmd
19