[Mobile] Fix The Build For Model Tracer (#84755)
Summary: Currently, the model tracer build is broken because of 2 reasons:
1. A few source files are missing, resulting in missing link time symbols
2. The `TRACING_BASED` flag isn't passed correctly from the command line (specified as an evnironment variable) as a CMake flag
Both these issues were fixed.
Test Plan: Ran this command: `USE_CUDA=0 TRACING_BASED=1 python setup.py develop --cmake`
and saw that the tracer binary was built at `build/bin/model_tracer` - also ran it to ensure that it can generate a YAML file.
Differential Revision: [D39391270](https://our.internmc.facebook.com/intern/diff/D39391270)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84755
Approved by: https://github.com/cccclai
diff --git a/build_variables.bzl b/build_variables.bzl
index 83853fa..718fcb2 100644
--- a/build_variables.bzl
+++ b/build_variables.bzl
@@ -552,6 +552,8 @@
"torch/csrc/jit/mobile/model_tracer/MobileModelRunner.cpp",
"torch/csrc/jit/mobile/model_tracer/OperatorCallTracer.cpp",
"torch/csrc/jit/mobile/model_tracer/KernelDTypeTracer.cpp",
+ "torch/csrc/jit/mobile/model_tracer/CustomClassTracer.cpp",
+ "torch/csrc/jit/mobile/model_tracer/BuildFeatureTracer.cpp",
]
torch_mobile_core = [
