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