From ba6fcc4eae2bdcf6e40d25c99b8150509adf5cba Mon Sep 17 00:00:00 2001
From: v-s-2 <22730518+v-s-2@users.noreply.github.com>
Date: Wed, 16 Aug 2023 19:52:12 +0000
Subject: [PATCH] [caffe2][SDT] Check whether `TORCH_DISABLE_SDT` macro is
 defined before referencing it (#107149)

Summary:
Some jobs in the next diff in stack (D48229150) fail with the following message:

```
stderr: In file included from xplat/caffe2/c10/cuda/CUDACachingAllocator.cpp:9:
xplat/caffe2/c10/util/static_tracepoint.h:4:6: error: 'TORCH_DISABLE_SDT' is not defined, evaluates to 0 [-Werror,-Wundef]
    !TORCH_DISABLE_SDT
```

When porting USDT macros to PyTorch in D47159249, I must have not hit a codepath which treated warnings as errors during testing.

This diff fixes the issue by first checking whether the `TORCH_DISABLE_SDT` macro is defined before trying to access it in the `static_tracepoint.h` header.

Test Plan:
Similar to D47159249, tested the following macro on test scripts with `libbpf` USDTs:
* `CAFFE_DISABLE_SDT`

Reviewed By: chaekit

Differential Revision: D48251736

Pull Request resolved: https://github.com/pytorch/pytorch/pull/107149
Approved by: https://github.com/chaekit
---
 c10/util/static_tracepoint.h | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/c10/util/static_tracepoint.h b/c10/util/static_tracepoint.h
index 019789cd46c5..4dee97306d14 100644
--- a/c10/util/static_tracepoint.h
+++ b/c10/util/static_tracepoint.h
@@ -1,7 +1,7 @@
 #pragma once
 
 #if defined(__ELF__) && (defined(__x86_64__) || defined(__i386__)) && \
-    !TORCH_DISABLE_SDT
+    !(defined(TORCH_DISABLE_SDT) && TORCH_DISABLE_SDT)
 
 #define TORCH_HAVE_SDT 1