| 34 | namespace cuda { |
| 35 | |
| 36 | char* GetTFExtraPTXOptions() { |
| 37 | static bool fetched = false; |
| 38 | static char* env = nullptr; |
| 39 | if (!fetched) { |
| 40 | env = getenv("TF_EXTRA_PTXAS_OPTIONS"); |
| 41 | fetched = true; |
| 42 | } |
| 43 | return env; |
| 44 | } |
| 45 | |
| 46 | #if defined(PLATFORM_WINDOWS) |
| 47 | port::StatusOr<std::vector<uint8>> CompilePtx(int device_ordinal, |