Merge pull request #1003 from NVIDIA/anuragd/extend_dla_support_on_orin

feat: Added compute capability for Orin

Author: narendasan

Diff for: core/lowering/register_trt_placeholder_ops.cpp

@@ -10,10 +10,7 @@ c10::AliasAnalysisKind aliasAnalysisFromSchema() {
 RegisterOperators trt_placeholder_ops_reg({
     /// Op marks a Tensor to be conveted from an Torch Tensor
     /// to a TRT constant Tensor
-    Operator(
-        "trt::const(Tensor val) -> Tensor",
-        [](Stack& stack) { /*noop*/ },
-        aliasAnalysisFromSchema()),
+    Operator("trt::const(Tensor val) -> Tensor", [](Stack& stack) { /*noop*/ }, aliasAnalysisFromSchema()),
 });
 
 } // namespace jit

Diff for: core/runtime/runtime.cpp

@@ -105,7 +105,7 @@ DeviceList get_available_device_list() {
 // SM Compute capability <Compute Capability, Device Name> map
 const std::unordered_map<std::string, std::string>& get_dla_supported_SMs() {
   // Xavier SM Compute Capability
-  static std::unordered_map<std::string, std::string> dla_supported_SM = {{"7.2", "Xavier"}};
+  static std::unordered_map<std::string, std::string> dla_supported_SM = {{"7.2", "Xavier"}, {"8.7", "Orin"}};
   return dla_supported_SM;
 }